RunTasking / app.py
wasmdashai's picture
Update app.py
7fef4b1 verified
raw
history blame
633 Bytes
import gradio as gr
GK=0
from transformers import AutoTokenizer,VitsModel
import torch
token=os.environ.get("key_")
tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)#.to(device)
def greet(name):
global GK
while True:
GK+=1
model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)#.to(device)
yield f"namber is {GK}"
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()