Fix NameError by adding os import in main.py

main.py

@@ -1,3 +1,4 @@
+import os
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from transformers import pipeline