Update app.py

app.py

@@ -10,21 +10,18 @@ import imageio
 import torch
 from custom_wan_pipeline import WanImageToVideoPipeline
 
+
 app = FastAPI()
 
-# ===== LOAD YOUR MODEL =====
 print("Loading WAN I2V model...")
 pipe = WanImageToVideoPipeline.from_pretrained(
-    ".",  # model files live in the repo root
+    ".",  
     torch_dtype=torch.float16
 ).to("cuda")
-
 pipe.enable_xformers_memory_efficient_attention()
 print("Model loaded.")
 
 
-# ===== Helpers =====
-
 def decode_image(b64_string):
     image_bytes = base64.b64decode(b64_string)
     return Image.open(io.BytesIO(image_bytes)).convert("RGB")
@@ -42,32 +39,28 @@ def frames_to_base64_mp4(frames):
     return base64.b64encode(video_bytes).decode()
 
 
-# ===== ROUTES =====
-
 @app.post("/video")
 async def video_route(body: dict):
     try:
         image_b64 = body["image"]
         prompt = body.get("prompt", "")
 
-        # Decode image
         image = decode_image(image_b64)
 
-        # Run WAN model
         output = pipe(image=image, prompt=prompt)
         frames = output.frames
 
-        # Convert frames → mp4 → base64
         video_b64 = frames_to_base64_mp4(frames)
 
         return JSONResponse({"video": video_b64})
+
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
 
-# HF Spaces launch
 def start():
     uvicorn.run(app, host="0.0.0.0", port=7860)
 
+
 if __name__ == "__main__":
     start()