Update app.py

app.py

@@ -1,38 +1,8 @@
-import os
-from dotenv import load_dotenv, find_dotenv
-_ = load_dotenv(find_dotenv())
-hf_api_key = os.environ['HF_API_KEY']
+from src.controllers.text_summarize_controller import loadGUI
 
-API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
 
+def main():
+    loadGUI()
 
-# Helper function
-import requests, json
-
-#Summarization endpoint
-def get_completion(inputs, parameters=None,ENDPOINT_URL=API_URL): 
-    headers = {
-      "Authorization": f"Bearer {hf_api_key}",
-      "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"))
-
-
-from transformers import pipeline
-
-model_end_point = "facebook/bart-large-cnn"
-get_completion = pipeline("summarization", model=model_end_point)
-
-
-import gradio as gr
-def summarize(input):
-    output = get_completion(input)
-    return output[0]['summary_text']
-    
-gr.close_all()
-demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
-demo.launch(share=True)
+if __name__ == '__main__':
+    main()