Update app.py

app.py

@@ -1,39 +1,13 @@
-import os
-import io
-from IPython.display import Image, display, HTML
-from PIL import Image
-import base64 
-from dotenv import load_dotenv, find_dotenv
-_ = load_dotenv(find_dotenv()) # read local .env file
-hf_api_key = os.environ['HF_API_KEY']
-# Helper function
-import requests, json
 
-#Summarization endpoint
-def get_completion(inputs, parameters=None,ENDPOINT_URL=os.environ['HF_API_SUMMARY_BASE']): 
-    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"))
-    import gradio as gr
+import gradio as gr
+
+from transformers import pipeline
+
+get_completion = pipeline("summarization", model="Falconsai/text_summarization")
 
 def summarize(input):
     output = get_completion(input)
     return output[0]['summary_text']
 
 gr.close_all()
-demo = gr.Interface(fn=summarize, 
-                    inputs=[gr.Textbox(label="Text to summarize", lines=6)],
-                    outputs=[gr.Textbox(label="Result", lines=3)],
-                    title="Text summarization with distilbart-cnn",
-                    description="Summarize any text using the `shleifer/distilbart-cnn-12-6` model under the hood!"
-                   )
-    
+demo = gr.Interface(fn=summarize, inputs="text", outputs="text")