Update main.py

main.py

@@ -1,13 +1,13 @@
 from fastapi import FastAPI, File, UploadFile, Form
-from huggingface_hub import InferenceClient
 import uvicorn
 from fastapi.responses import StreamingResponse
 import io
 import base64
+import requests
 
 app = FastAPI()
 
-client = InferenceClient("stabilityai/stable-video-diffusion-img2vid-xt-1-1-tensorrt")
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-video-diffusion-img2vid-xt-1-1-tensorrt"
 
 @app.post("/generate_video/")
 async def generate_video_api(
@@ -24,18 +24,24 @@ async def generate_video_api(
     # Encode the image to base64
     image_base64 = base64.b64encode(image_content).decode('utf-8')
 
-    # Generate the video
-    video = client.image_to_video(
-        image=image_base64,
-        fps=fps,
-        num_frames=num_frames,
-        motion_bucket_id=motion_bucket_id,
-        cond_aug=cond_aug,
-        seed=seed
-    )
-    
+    # Prepare the payload
+    payload = {
+        "inputs": image_base64,
+        "parameters": {
+            "fps": fps,
+            "num_frames": num_frames,
+            "motion_bucket_id": motion_bucket_id,
+            "cond_aug": cond_aug,
+            "seed": seed
+        }
+    }
+
+    # Make the API call
+    response = requests.post(API_URL, json=payload)
+    response.raise_for_status()
+
     # Return the video as a streaming response
-    return StreamingResponse(io.BytesIO(video), media_type="video/mp4")
+    return StreamingResponse(io.BytesIO(response.content), media_type="video/mp4")
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)