Update app.py

app.py

@@ -1,16 +1,25 @@
 import os
 import gradio as gr
+from gradio_client import Client
 
 MY_HF_TOKEN_KEY = os.environ["MY_HF_TOKEN_KEY"]
 
-iface = gr.load(name="Ghana-NLP/Khaya-chat-bot",hf_token=MY_HF_TOKEN_KEY,src ='spaces')
+#iface = gr.load(name="Ghana-NLP/Khaya-chat-bot",hf_token=MY_HF_TOKEN_KEY,src ='spaces')
+client = Client("Ghana-NLP/Khaya-chat-bot",hf_token=MY_HF_TOKEN_KEY,src ='spaces')
+
 with gr.Blocks() as demo:
+    title = gr.Markdown(
+    """
+    # African Language Chatbot (Khaya)
+    SELECT YOUR LANGUAGE BELOW!
+    """)    
+    language_selector = gr.Dropdown(["Twi","Dagbani","Ewe","Fante","Ga","Gurene","Kikuyu","Kimeru","Luo","Yoruba","Hausa","Swahili"],value="Twi", label="", info="LANGUAGE:")
     with gr.Row():
-        output = gr.Text(label="Conversation")
+        output = gr.Text(label="Conversation:")
     with gr.Row():
-        prompt = gr.Text(label="Enter Prompt In Your Local Language")
+        prompt = gr.Text(label="Enter Text In Your Local Language:")
     btn = gr.Button("Chat")
-    btn.click(iface, inputs=[prompt], outputs=[output])
+    btn.click(client.predict, inputs=[prompt,language_selector], outputs=[output])
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()