MihirRajeshPanchal commited on
Commit
162d0ff
1 Parent(s): 791f42a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,4 +1,6 @@
1
  import gradio as gr
 
 
2
 
3
  from diffusers import DiffusionPipeline
4
 
@@ -6,13 +8,16 @@ from diffusers import DiffusionPipeline
6
  pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
7
 
8
  # Define the function for Gradio to use
9
- def generate_video(input_image_path):
 
 
 
10
  output_video_path = "output_video.mp4"
11
- pipeline(input_image_path, output_video_path)
12
  return output_video_path
13
 
14
  # Create the Gradio interface
15
  iface = gr.Interface(fn=generate_video, inputs="image", outputs="file")
16
 
17
- # Launch the Gradio app
18
- iface.launch()
 
1
  import gradio as gr
2
+ from PIL import Image
3
+ import numpy as np
4
 
5
  from diffusers import DiffusionPipeline
6
 
 
8
  pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
9
 
10
  # Define the function for Gradio to use
11
+ def generate_video(input_image):
12
+ # Convert NumPy array to PIL Image
13
+ input_image_pil = Image.fromarray((input_image * 255).astype(np.uint8))
14
+
15
  output_video_path = "output_video.mp4"
16
+ pipeline(input_image_pil, output_video_path)
17
  return output_video_path
18
 
19
  # Create the Gradio interface
20
  iface = gr.Interface(fn=generate_video, inputs="image", outputs="file")
21
 
22
+ # Launch the Gradio app with sharing enabled
23
+ iface.launch(share=True)