Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|