focusit commited on
Commit
d215242
·
1 Parent(s): 973ddbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -4,9 +4,24 @@
4
  # gr.Interface.load("models/deepset/roberta-base-squad2").launch()
5
 
6
  import gradio
 
 
 
 
 
 
 
 
7
 
8
  def my_inference_function(name):
9
- return "Hello " + name + "!"
 
 
 
 
 
 
 
10
 
11
  gradio_interface = gradio.Interface(
12
  fn = my_inference_function,
 
4
  # gr.Interface.load("models/deepset/roberta-base-squad2").launch()
5
 
6
  import gradio
7
+ import requests
8
+
9
+ API_URL = "https://api-inference.huggingface.co/models/deepset/roberta-base-squad2"
10
+ headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
11
+
12
+ def query(payload):
13
+ response = requests.post(API_URL, headers=headers, json=payload)
14
+ return response.json()
15
 
16
  def my_inference_function(name):
17
+ return query({
18
+ "inputs": {
19
+ "question": name,
20
+ "context": "My name is Clara and I live in Berkeley."
21
+ },
22
+ })
23
+
24
+
25
 
26
  gradio_interface = gradio.Interface(
27
  fn = my_inference_function,