import gradio as gr
from transformers import VitsModel, AutoTokenizer
import torch



def greet(name):
    model = VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba-v2",cache_dir=".")
    tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba-v2",cache_dir=".")
    return "Hello " + name + "!!"

demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()