Update app.py

app.py

@@ -1,5 +1,3 @@
-import os
-
 from groq import Groq 
 import gradio as gr 
 
@@ -59,44 +57,29 @@ def response_from_mistral(query):
 #     )
 # iface.launch()
 
+def chat_with_models(text):
+    llama_response = response_from_llam3(text)
+    mistral_response =response_from_mistral(text)
+    
+    return llama_response, mistral_response
+
+
 with gr.Blocks() as demo:
     gr.Markdown("<h1>🚀 Mistral 7B vs LLama3 8B 🦙</h1>")
     gr.Markdown("<h3> 🕹️ Type your questions or prompts related to Ayurveda and see how each model responds to the same input 👾 </h3>")
-    
-    example_questions = [
-        'What is importance of fasting according to Ayurveda?',
-        'What are the medicinal values of Tulsi?',
-        'What are the three different doshas?',
-        'What is the ideal diet according to Ayurveda?'
-    ]
-    
-    input_text = gr.Textbox(label="Enter your prompt here:", placeholder="Type something...", lines=2)
-    example_dropdown = gr.Dropdown(label="Select an Example Question", choices=example_questions)
-    submit_button = gr.Button("Submit")
-    
+    with gr.Row():
+        input_text = gr.Textbox(label="Enter your prompt here:", placeholder="Type something...", lines=2)
+        submit_button = gr.Button("Submit")
     output_llama = gr.Textbox(label="Llama 3 8B 👾", placeholder="", lines=10, interactive=False)
     output_mistral = gr.Textbox(label="Mistral 7B 🌠", placeholder="", lines=10, interactive=False)
+    examples=[
+            ['What is importance of fasting according to Ayurveda?'],
+            ['What are the medicinal values of Tusli?'],
+            ['What are the three different doshas?'],
+            ['What is the ideal diet according to ayurveda?']
+            ],
     
-    def set_example_question(choice):
-        input_text.value = choice
-    
-    example_dropdown.change(set_example_question)
-    
-    def chat_with_models():
-        llama_response = response_from_llam3(input_text.value)
-        mistral_response = response_from_mistral(input_text.value)
-        output_llama.value = llama_response
-        output_mistral.value = mistral_response
-    
-    submit_button.click(chat_with_models)
+    submit_button.click(fn=chat_with_models, inputs=input_text, outputs=[output_llama,output_mistral])
 
 if __name__ == "__main__":
-    demo.launch(share=True)
-
-
-
-
-
-
-
-
+    demo.launch()