Update app.py

app.py

@@ -7,6 +7,10 @@ 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')
 client = Client("Ghana-NLP/Khaya-chat-bot",hf_token=MY_HF_TOKEN_KEY)
 
+def generate(prompt,language):
+    output = client.predict(prompt,language)
+    return output
+
 with gr.Blocks() as demo:
     title = gr.Markdown(
     """
@@ -19,7 +23,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         prompt = gr.Text(label="Enter Text In Your Local Language:")
     btn = gr.Button("Chat")
-    btn.click(client.predict, inputs=[prompt,language_selector], outputs=[output])
+    btn.click(generate, inputs=[prompt,language_selector], outputs=[output])
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()