wasmdashai commited on
Commit
59cd719
·
verified ·
1 Parent(s): c887ba1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def get_model(name_model):
26
  zero = torch.Tensor([0]).cuda()
27
  print(zero.device) # <-- 'cpu' 🤔
28
  import torch
29
- @spaces.GPU()
30
  def modelspeech(text,name_model):
31
 
32
 
@@ -49,4 +49,5 @@ model_choices = gr.Dropdown(
49
  value="wasmdashai/vtk",
50
  )
51
  demo = gr.Interface(fn=modelspeech, inputs=["text",model_choices], outputs=["audio"])
 
52
  demo.launch()
 
26
  zero = torch.Tensor([0]).cuda()
27
  print(zero.device) # <-- 'cpu' 🤔
28
  import torch
29
+ @spaces.GPU(duration=120)
30
  def modelspeech(text,name_model):
31
 
32
 
 
49
  value="wasmdashai/vtk",
50
  )
51
  demo = gr.Interface(fn=modelspeech, inputs=["text",model_choices], outputs=["audio"])
52
+ demo.queue()
53
  demo.launch()