Update app.py

app.py

@@ -1,15 +1,18 @@
 import gradio as gr
 import requests
 
-
 API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
 headers = {"Authorization": "Bearer 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"}
+
 API_URL2 = "https://api-inference.huggingface.co/models/valhalla/longformer-base-4096-finetuned-squadv1"
 headers2 = {"Authorization": "Bearer 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"}
+
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
-def query2(payload):
+
+def query2(question, context):  # Accept two arguments
+    payload = {"question": question, "context": context}
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
@@ -22,3 +25,4 @@ iface = gr.Interface(
 )
 
 iface.launch()
+