Update app.py

app.py

@@ -7,30 +7,29 @@ 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(question):
-    payload = {"question": "what is the context of "+question+" : "}
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()["answer"]
+current_model = 1  # Default to the first model
 
-def query2(question, context):
-    payload = {"question": question, "context": context}
-    response = requests.post(API_URL2, headers=headers2, json=payload)
-    return response.json()["answer"]
+def switch_model():
+    global current_model
+    current_model = 1 if current_model == 2 else 2
 
-iface = gr.Interface(
-    fn=query2,
-    inputs=[gr.Textbox("question"), gr.Textbox("context")],
-    outputs=gr.Textbox("answer"),
-    title="AI Interface 2",
-    description="Ask the AI model anything!",
-)
+def query(question, context=""):
+    if current_model == 1:
+        payload = {"question": question, "context": context}
+        response = requests.post(API_URL2, headers=headers2, json=payload)
+        return response.json()["answer"]
+    else:
+        payload = {"question": "what is the context of the question: "+question+" :"}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        return response.json()["answer"]
 
-iface2 = gr.Interface(
+iface = gr.Interface(
     fn=query,
-    inputs=[gr.Textbox("question")],
+    inputs=[gr.Textbox("question"), gr.Textbox("context"), gr.Button("Switch Model", switch_model)],
     outputs=gr.Textbox("answer"),
     title="AI Interface",
     description="Ask the AI model anything!",
 )
 
-gr.launch_in_debugger([iface, iface2])
+iface.launch()
+