fix: write uploaded file to /tmp for HF compatibility

main.py

@@ -6,6 +6,7 @@ from fastapi.staticfiles import StaticFiles
 import shutil
 from pathlib import Path
 import uvicorn
+import os
 
 from vit_captioning.generate import CaptionGenerator
 
@@ -46,9 +47,10 @@ caption_generator = CaptionGenerator(
     quantized=False
 )
 
+@app.post("/generate")
 @app.post("/generate")
 async def generate(file: UploadFile = File(...)):
-    temp_file = f"temp_{file.filename}"
+    temp_file = os.path.join("/tmp", file.filename)
     with open(temp_file, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)