LPX55 commited on
Commit
b6c95c0
·
verified ·
1 Parent(s): 3ab55c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import safetensors.torch
3
  import torchvision.transforms.v2 as transforms
@@ -99,7 +100,7 @@ def resize_image_to_bucket(image: Union[Image.Image, np.ndarray], bucket_reso: T
99
  return image
100
 
101
 
102
-
103
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
104
  # Debugging print statements
105
  print(f"Frame 1 Type: {type(frame1)}")
 
1
+ import spaces
2
  import gradio as gr
3
  import safetensors.torch
4
  import torchvision.transforms.v2 as transforms
 
100
  return image
101
 
102
 
103
+ @spaces.GPU()
104
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
105
  # Debugging print statements
106
  print(f"Frame 1 Type: {type(frame1)}")