rahul7star commited on
Commit
ad20238
·
verified ·
1 Parent(s): fc6bdf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import sys
7
  import json
8
  import warnings
9
  from datetime import datetime
10
-
11
  import gradio as gr
12
  warnings.filterwarnings('ignore')
13
 
@@ -515,7 +515,7 @@ def run_graio_demo(args):
515
  )
516
 
517
 
518
-
519
  def generate_video(img2vid_image, vid2vid_vid, task_mode, img2vid_prompt, n_prompt, img2vid_audio_1, img2vid_audio_2,
520
  sd_steps, seed, text_guide_scale, audio_guide_scale, mode_selector, tts_text, resolution_select, human1_voice, human2_voice):
521
  input_data = {}
 
7
  import json
8
  import warnings
9
  from datetime import datetime
10
+ import spaces
11
  import gradio as gr
12
  warnings.filterwarnings('ignore')
13
 
 
515
  )
516
 
517
 
518
+ @spaces.GPU(duration=60)
519
  def generate_video(img2vid_image, vid2vid_vid, task_mode, img2vid_prompt, n_prompt, img2vid_audio_1, img2vid_audio_2,
520
  sd_steps, seed, text_guide_scale, audio_guide_scale, mode_selector, tts_text, resolution_select, human1_voice, human2_voice):
521
  input_data = {}