Update app.py

app.py

@@ -27,7 +27,7 @@ headers = {"Authorization": f"Bearer {api_key}"}
 #     return (df_state.sr(),np.flipud(enhanced))
 
 
-def query(text):
+def query(text,API_URL):
   payload={"inputs": text}
   response = requests.post(API_URL, headers=headers, json=payload)
   return response.content
@@ -43,11 +43,17 @@ def   get_answer_ai(text):
           AI=create_chat_session()
           response = AI.send_message(text)
           return response.text
-          
-def reverse_audio(text,intensity):
+model_choices = gr.Dropdown(
+    choices=["asg2024/vits-ar-sa", "other_model1", "other_model2"],  # Replace with your model names
+    label="اختر النموذج",
+    value="asg2024/vits-ar-sa"  # Default selection
+)
+     
+def reverse_audio(text,model_choice):
+    API_URL = f"https://api-inference.huggingface.co/models/{model_choice}" 
     text_answer=get_answer_ai(text)
-    dataq = query(text)
-    dataa = query(text_answer)
+    dataq = query(text,API_URL)
+    dataa = query(text_answer,API_URL)
     
     # data_enhanc = remove_noise_enhance(data)
 
@@ -57,7 +63,7 @@ def reverse_audio(text,intensity):
 
 demo = gr.Interface(
     fn=reverse_audio,
-    inputs=["text","slider"],
+    inputs=["text",model_choices],
     outputs=["audio","audio","text"]
 )