Update main.py

main.py

@@ -1,5 +1,4 @@
-from fastapi import FastAPI, File, UploadFile
-from pydantic import BaseModel
+from fastapi import FastAPI, File, UploadFile, Form
 from huggingface_hub import InferenceClient
 import uvicorn
 from fastapi.responses import StreamingResponse
@@ -10,15 +9,15 @@ app = FastAPI()
 
 client = InferenceClient("stabilityai/stable-video-diffusion-img2vid-xt-1-1-tensorrt")
 
-class Item(BaseModel):
-    fps: int = 7
-    num_frames: int = 14
-    motion_bucket_id: int = 127
-    cond_aug: float = 0.02
-    seed: int = 0
-
 @app.post("/generate_video/")
-async def generate_video_api(item: Item, file: UploadFile = File(...)):
+async def generate_video_api(
+    file: UploadFile = File(...),
+    fps: int = Form(7),
+    num_frames: int = Form(14),
+    motion_bucket_id: int = Form(127),
+    cond_aug: float = Form(0.02),
+    seed: int = Form(0)
+):
     # Read the uploaded image file
     image_content = await file.read()
     
@@ -28,11 +27,11 @@ async def generate_video_api(item: Item, file: UploadFile = File(...)):
     # Generate the video
     video = client.image_to_video(
         image=image_base64,
-        fps=item.fps,
-        num_frames=item.num_frames,
-        motion_bucket_id=item.motion_bucket_id,
-        cond_aug=item.cond_aug,
-        seed=item.seed
+        fps=fps,
+        num_frames=num_frames,
+        motion_bucket_id=motion_bucket_id,
+        cond_aug=cond_aug,
+        seed=seed
     )
     
     # Return the video as a streaming response