Update app.py

app.py

@@ -9,16 +9,16 @@ headers2 = {"Authorization": "Bearer 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"}
 def query(payload):
     payload2 = "whats the context of the question " + paylaod + " :"
     response = requests.post(API_URL, headers=headers, json=payload2)
-    return response.json() #["answer"]
+    return response.json()['answer']
 def query2(q, c):
     payload = {"question": q, "context": c}
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()['answer']
 
 iface = gr.Interface(
-    fn=query2,
-    inputs=[gr.Textbox("question"), gr.Textbox("context")],
-    outputs=gr.Textbox("answer"),
+    fn=query,
+    inputs=[gr.Textbox("question")],
+    outputs=gr.Textbox("context"),
     title="AI Interface",
     description="Ask the AI model anything!",
 )