Update main.py

main.py

@@ -11,6 +11,8 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 from fastapi.responses import FileResponse
 
+app = FastAPI()
+
 # Thiết lập logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -139,4 +141,8 @@ async def generate_video(request: VideoRequest):
 # Endpoint kiểm tra trạng thái
 @app.get("/")
 async def root():
-    return {"message": "FastAPI AnimateDiff-Lightning API on Hugging Face Spaces"}
+    return {"message": "FastAPI AnimateDiff-Lightning API on Hugging Face Spaces"}
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 7860))  # Nếu PORT không có thì mặc định 7860
+    uvicorn.run("main:app", host="0.0.0.0", port=port, reload=False)