multimodalart HF Staff commited on
Commit
379cab9
·
verified ·
1 Parent(s): bd6580c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,9 +3,11 @@ import time
3
  import numpy as np
4
  import gradio as gr
5
  import colorsys
 
6
 
7
  from fastrtc import WebRTC, get_turn_credentials
8
 
 
9
  def video_generation_handler(prompt):
10
  num_frames = 120
11
  width, height = 832, 480
 
3
  import numpy as np
4
  import gradio as gr
5
  import colorsys
6
+ import spaces
7
 
8
  from fastrtc import WebRTC, get_turn_credentials
9
 
10
+ @spaces.GPU
11
  def video_generation_handler(prompt):
12
  num_frames = 120
13
  width, height = 832, 480