[DOWNLOAD TEST]

The-Swarm-Corporation · Apr 16, 2024 · e844671 · e844671
1 parent 8fab556
commit e844671
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/text_to_video.py b/text_to_video.py
@@ -140,6 +140,11 @@ async def create_chat_completion(
         # logger.info(f"Response: {out}")
         logger.info(f"Downloading the file: {response}")
 
+        FileResponse(
+            response,
+            media_type="image/gif",
+            filename=request.output_path
+        )
 
         return JSONResponse(content=log.model_dump(), status_code=200)
     except Exception as e: