Spaces:
Sleeping
Sleeping
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(q): | |
payload2 = "whats the context of the question " + q + " :" | |
response = requests.post(API_URL, headers=headers, json=payload2) | |
return response.json()['answer'] | |
def query2(q, c): | |
payload = {"question": q, "context": c} | |
response = requests.post(API_URL2, headers=headers2, json=payload) | |
return response.json()['answer'] | |
iface = gr.Interface( | |
fn=query, | |
inputs=[gr.Textbox("question")], | |
outputs=gr.Textbox("context"), | |
title="AI Interface", | |
description="Ask the AI model anything!", | |
) | |
iface.launch() |