imagen / app.py
focusit's picture
Update app.py
c4bf75b
raw
history blame
718 Bytes
# import gradio as gr
# gr.Interface.load("models/deepset/roberta-base-squad2").launch()
import gradio
import requests
import os
API_URL = "https://api-inference.huggingface.co/models/deepset/roberta-base-squad2"
headers = {"Authorization": os.environ.getattribute("Authorization")}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
def my_inference_function(name):
return query({
"inputs": {
"question": name,
"context": "My name is Clara and I live in Berkeley."
},
})
gradio_interface = gradio.Interface(
fn = my_inference_function,
inputs = "text",
outputs = "text"
)
gradio_interface.launch()