Update app.py

app.py

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