Upload 3 files

src/constants/constants.py

@@ -0,0 +1 @@
+CNN_API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"

src/controllers/text_summarize_controller.py

@@ -0,0 +1,16 @@
+
+import gradio as GUI
+from transformers import pipeline
+from ..services.text_summarize_service import completion_message
+
+completion_message = pipeline("summarization", model='facebook/bart-large-cnn')
+
+
+def summarize(input):
+    output = completion_message(input)
+    return output[0]['summary_text']
+    
+def loadGUI():
+    GUI.close_all()
+    demo = GUI.Interface(fn=summarize, inputs="text", outputs="text")
+    demo.launch(share=True)

src/services/text_summarize_service.py

@@ -0,0 +1,20 @@
+import os
+import requests, json
+from dotenv import load_dotenv, find_dotenv
+from ..constants.constants import CNN_API_URL
+
+
+load_dotenv(find_dotenv())
+huggingFaceAPIToken = os.environ['HF_API_KEY']
+
+
+def completion_message(inputs, parameters=None,ENDPOINT_URL=CNN_API_URL): 
+    headers = {
+      "Authorization": f"Bearer {huggingFaceAPIToken}",
+      "Content-Type": "application/json"
+    }
+    data = { "inputs": inputs }
+    if parameters is not None:
+        data.update({"parameters": parameters})
+    response = requests.request("POST",ENDPOINT_URL, headers=headers,data=json.dumps(data))
+    return json.loads(response.content.decode("utf-8"))