Update app.py

app.py

@@ -15,17 +15,17 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def inference_ui(question):
+def inference_ui(question, context_input):
     output = query({
-        "inputs": f"context for '{question}' is:",
+        "inputs": f"context for '{question}' is: {context_input}",
     })
     output2 = query2({
         "inputs": {
             "question": question,
-            "context": output
+            "context": output['context']  # Ensure that 'context' is a string
         },
     })
     return output2
 
-iface = gr.Interface(fn=inference_ui, inputs="text", outputs="text")
+iface = gr.Interface(fn=inference_ui, inputs=["text", "text"], outputs="text")
 iface.launch()