Update app.py

app.py

@@ -64,7 +64,7 @@ pipe.to("cuda")
 
 max_64_bit_int = 2**63 - 1
 
-@spaces.GPU(duration=180)
+@spaces.GPU(enable_queue=True, duration=240)
 def generate_video(
     image: Image,
     seed: int,
@@ -97,17 +97,18 @@ def generate_video(
     torch.manual_seed(seed)
 
     # Read the content of the video file and encode it to base64
-    with open(video_path, "rb") as video_file:
-        video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
+    # with open(video_path, "rb") as video_file:
+    #     video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
 
     # Prepend the appropriate data URI header with MIME type
-    video_data_uri = 'data:video/mp4;base64,' + video_base64
+    # video_data_uri = 'data:video/mp4;base64,' + video_base64
     
     # clean-up (otherwise there is a risk of "ghosting", eg. someone seeing the previous generated video",
     # of one of the steps go wrong)
-    os.remove(video_path)
+    # os.remove(video_path)
     
-    return video_data_uri
+    # return video_data_uri
+    return video_path
 
 def resize_image(image, output_size=(1024, 576)):
     # Calculate aspect ratios
@@ -142,26 +143,39 @@ def resize_image(image, output_size=(1024, 576)):
 
 
 css = """
-
 img, video {
     max-height: 400px;
     object-fit: contain;
 }
+
+video {
+    margin: 0 auto
+}
 """
 
-with gr.Blocks(css=css) as demo:
-    image = gr.Image(label="Upload your image", type="pil")
-    generate_btn = gr.Button("Generate")
-    base64_out = gr.Textbox(label="Base64 Video")
-    seed = gr.Slider(label="Seed", value=42, randomize=False, minimum=0, maximum=max_64_bit_int, step=1)
-    motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, minimum=1, maximum=255)
-    fps_id = gr.Slider(label="Frames per second", info="The length of your video in seconds will be 25/fps", value=6, minimum=5, maximum=30)
+with gr.Blocks(css=css) as SVD_XT_1_1:
+    with gr.Row():
+        with gr.Column():
+            image = gr.Image(label="Upload your image", type="pil")
+            generate_btn = gr.Button("Generate")
+            # base64_out = gr.Textbox(label="Base64 Video")
+            seed = gr.Slider(label="Seed", value=42, randomize=False, minimum=0, maximum=max_64_bit_int, step=1)
+            motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, minimum=1, maximum=255)
+            fps_id = gr.Slider(label="Frames per second", info="The length of your video in seconds will be 25/fps", value=6, minimum=5, maximum=30)
+
+        with gr.Column():
+            video_out = gr.Video(
+                autoplay=True,
+                # height=512,
+                # width=512,
+                # elem_id="video_output"
+            )
        
     generate_btn.click(
         fn=generate_video,
         inputs=[image, seed, motion_bucket_id, fps_id],
-        outputs=base64_out,
+        outputs=video_out,
         api_name="run"
     )
 
-demo.launch()
+SVD_XT_1_1.launch()