Update main.py

main.py

@@ -10,7 +10,8 @@ app = FastAPI()
 cache_dir = './cache'
 
 # Create the cache directory if it doesn't exist
-os.makedirs(cache_dir, exist_ok=True)
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir, exist_ok=True)
 
 # Set the TRANSFORMERS_CACHE environment variable to the cache directory
 os.environ['TRANSFORMERS_CACHE'] = cache_dir
@@ -60,4 +61,4 @@ async def generate_text(item: Item):
         raise HTTPException(status_code=500, detail=f"An error occurred: {e}")
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)