Create app.py

app.py

@@ -0,0 +1,41 @@
+import os
+
+hf_api_url = os.environ('HF_API_URL')
+hf_api_key = os.environ('HF_API_KEY')
+
+### Helper Functions
+import requests, json
+
+headers = {
+      "Authorization": f"Bearer {hf_api_key}",
+      "Content-Type": "application/json"
+    }
+
+def get_completion(inputs, parameters = None):
+
+    payload = {"inputs" : inputs}
+
+    if paramaters is not None:
+        payload.update({"parameters" : parameters})
+    
+    response = requests.post(hf_api_url, headers = headers, data = json.dumps(payload))
+
+    return json.loads(response.content.decode("utf-8"))
+
+### Gradio App
+import gradio as gr
+
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output} 
+
+examples = [
+    "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(),
+             examples = examples)
+
+demo.launch()