Update main.py

main.py

@@ -8,6 +8,7 @@ from safetensors.torch import load_file
 from PIL import Image
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
+from fastapi.responses import FileResponse
 import uvicorn
 
 app = FastAPI()
@@ -90,7 +91,7 @@ def generate_image(request: GenerateImageRequest):
     path = f"/tmp/{name}.mp4"
     export_to_video(output.frames[0], path, fps=10)
     
-    return {"video_path": path}
+    return FileResponse(path, media_type="video/mp4", filename=f"{name}.mp4")
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)