Spaces:

wasmdashai
/

LAHJA-AI

Sleeping

wasmdashai commited on Aug 21, 2024

Commit

5d6531a

verified ·

1 Parent(s): 59cd719

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import os
 token=os.environ.get("key_")
 tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vtk",token=token)
 models= {}
 def  get_model(name_model):
     global models
     if name_model in   models:
@@ -26,7 +26,7 @@ def  get_model(name_model):
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
 import torch
-@spaces.GPU(duration=120)
 def   modelspeech(text,name_model):

 token=os.environ.get("key_")
 tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vtk",token=token)
 models= {}
+@spaces.GPU
 def  get_model(name_model):
     global models
     if name_model in   models:
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
 import torch
+@spaces.GPU
 def   modelspeech(text,name_model):