Update app.py

app.py

@@ -9,12 +9,12 @@ headers2 = {"Authorization": "Bearer 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"}
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    return response.json()["answer"]
 
 def query2(question, context):  # Accept two arguments
     payload = {"question": question, "context": context}
     response = requests.post(API_URL2, headers=headers2, json=payload)
-    return response.json()
+    return response.json()["answer"]
 
 iface = gr.Interface(
     fn=query2,