Spaces:
Paused
Paused
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()
|