Create main.py

main.py

@@ -0,0 +1,36 @@
+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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"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+API_URL2 = "https://api-inference.huggingface.co/models/valhalla/longformer-base-4096-finetuned-squadv1"
+headers2 = {"Authorization": "Bearer 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"}
+
+def query2(payload):
+    response = requests.post(API_URL2, headers=headers2, json=payload)
+    return response.json()
+
+def get_context(question):
+    payload = {"inputs": question}
+    result = query(payload)
+    return result['answer']
+
+def ask_ai(question, context):
+    payload = {"question": question, "context": context}
+    result = query2(payload)
+    return result['answer']
+
+iface = gr.Interface(
+    fn=ask_ai,
+    inputs=[gr.Textbox("Question", default=""), gr.Button("get_context", get_context)],
+    outputs=gr.Textbox("Answer"),
+    live=True,
+    capture_session=True,
+)
+
+iface.launch()