Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 import uvicorn
 from fastapi.responses import StreamingResponse
 import io
-import requests
+import base64
 from PIL import Image
 import ffmpeg
 import tempfile
@@ -27,29 +27,35 @@ async def generate_video_api(
         image_content = await file.read()
         image = Image.open(io.BytesIO(image_content))
 
-        # Convert image to bytes
-        img_byte_arr = io.BytesIO()
-        image.save(img_byte_arr, format='PNG')
-        img_byte_arr = img_byte_arr.getvalue()
+        # Convert image to base64
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode()
 
         # Generate video frames using the stable-video-diffusion model
-        video_frames = client.post(
-            "video-generation",
-            files=[("image", img_byte_arr)],
-            data={
-                "num_inference_steps": 25,
-                "num_frames": num_frames,
-                "motion_bucket_id": motion_bucket_id,
-                "cond_aug": cond_aug,
-                "seed": seed
+        response = client.post(
+            json={
+                "inputs": img_str,
+                "parameters": {
+                    "num_inference_steps": 25,
+                    "num_frames": num_frames,
+                    "motion_bucket_id": motion_bucket_id,
+                    "cond_aug": cond_aug,
+                    "seed": seed
+                }
             }
         )
 
+        # Check if the response is a list of images
+        if not isinstance(response, list):
+            raise ValueError(f"Unexpected response from the model: {response}")
+
         # Create a temporary directory
         with tempfile.TemporaryDirectory() as tmpdir:
             # Save frames as temporary files
             frame_files = []
-            for i, frame in enumerate(video_frames):
+            for i, frame_data in enumerate(response):
+                frame = Image.open(io.BytesIO(base64.b64decode(frame_data)))
                 frame_file = os.path.join(tmpdir, f"frame_{i:03d}.png")
                 frame.save(frame_file)
                 frame_files.append(frame_file)