LTTEAM commited on
Commit
213413f
·
verified ·
1 Parent(s): 49996f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- import spaces
4
  import numpy as np
5
  import random
6
  import os
@@ -78,7 +78,7 @@ def calc_new_dims(w, h):
78
  def get_duration(*args, duration_ui=0, **kwargs):
79
  return 75 if duration_ui > 7 else 60
80
 
81
- @spaces.GPU(duration=get_duration)
82
  def generate(prompt, neg_prompt, img_path, vid_path,
83
  height, width, mode, duration_ui, frames_to_use,
84
  seed, rand_seed, cfg_scale, improve_tex, device_choice,
@@ -254,4 +254,4 @@ with gr.Blocks(css=css) as demo:
254
  outputs=[out_vid, seed_state])
255
 
256
  if __name__ == "__main__":
257
- demo.queue().launch(debug=True, share=False)
 
1
  import gradio as gr
2
  import torch
3
+
4
  import numpy as np
5
  import random
6
  import os
 
78
  def get_duration(*args, duration_ui=0, **kwargs):
79
  return 75 if duration_ui > 7 else 60
80
 
81
+
82
  def generate(prompt, neg_prompt, img_path, vid_path,
83
  height, width, mode, duration_ui, frames_to_use,
84
  seed, rand_seed, cfg_scale, improve_tex, device_choice,
 
254
  outputs=[out_vid, seed_state])
255
 
256
  if __name__ == "__main__":
257
+ demo.queue().launch(debug=True, share=True)