Update app.py

app.py

@@ -20,22 +20,22 @@ def get_completion(inputs, parameters = None):
     
     response = requests.post(hf_api_url, headers = headers, data = json.dumps(payload))
 
-    return json.loads(response.content.decode("utf-8"))
+    return response.json()
 
 ### Gradio App
 import gradio as gr
 
 def ner(text):
     output = get_completion(text)
-    return {"text": text, "entities": output} 
+    return {"text" : text, "entities" : output} 
 
 examples = [
-    "Does Chicago have any stores and does Joe live here?",
+    "Does Chicago have any stores and does Joe live here?"
 ]
 
-demo = gr.Interface(fn = ner,
-             inputs = gr.Textbox(placeholder="Enter sentence here..."), 
-             outputs = gr.HighlightedText(),
+demo = gr.Interface(ner,
+             gr.Textbox(placeholder="Enter sentence here..."), 
+             gr.HighlightedText(),
              examples = examples)
 
 demo.launch()