Update app.py

app.py

@@ -6,6 +6,7 @@ from diffusers.utils import export_to_video
 import io
 from fastapi.responses import StreamingResponse
 import uvicorn
+import torch
 
 app = FastAPI()