Add Application File

Text_Summarization.py

@@ -0,0 +1,25 @@
+import requests
+import json
+import os
+
+def get_completion(inputs, parameters=None):
+    headers = {
+        "Authorization":f"Bearer {api_key}",
+        "Content-Type":"application/json"
+    }
+    data = {"inputs": inputs}
+    if parameters is not None:
+        data['parameters'] = parameters
+    response = requests.request("POST",
+                                endpoint_url,
+                                headers=headers,
+                                data=json.dumps(data)
+                                )
+    return json.loads(response.content.decode("utf-8"))
+
+def summarize(input):
+    output = get_completion(input)
+    return output[0]['summary_text']
+
+api_key = os.environ['HF_TOKEN']
+endpoint_url = os.environ['API_URL']

app.py

@@ -0,0 +1,6 @@
+import gradio as gr
+from Text_Summarization import summarize
+
+gr.close_all()
+demo = gr.Interface(fn=summarize, inputs='text', outputs='text')
+demo.launch()