Update app.py

app.py

@@ -8,12 +8,22 @@ tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=tok
 #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   modelspeech(texts):
+     
+    
+    
+     inputs = tokenizer(texts, return_tensors="pt")#.cuda()
 
-def greet(name):
+     wav = model_vits(input_ids=inputs["input_ids"]).waveform#.detach()
+          # display(Audio(wav, rate=model.config.sampling_rate))
+     return  model_vits.config.sampling_rate,wav#remove_noise_nr(wav)
+def greet(id):
     global GK 
+    b=int(id)
     while True:
         GK+=1
-        model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)#.to(device)
+        texts=['السلام عليكم']*b
+        out=modelspeech(texts)
         yield f"namber is {GK}"
 
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")