Update app.py

app.py

@@ -1,5 +1,8 @@
 # Import libraries
 import gradio as gr
+from transformers import AutoModel, AutoProcessor
+import torch
+import requests
 from transformers import pipeline
 
 # Import a specific model
@@ -9,9 +12,9 @@ model_name = "facebook/bart-large-cnn"
 summarizer = pipeline("summarization", model=model_name)
 
 # Define a function that takes a text and returns a summary
-def summarize(text):
-  summary = summarizer(text, max_length=150, min_length=40, do_sample=False)[0]
-  return summary["summary_text"]
+def summarize(input_text):
+  summary = summarizer(input_text, max_length=150, min_length=40, do_sample=False)
+  return summary[0]["summary_text"]
 
 # Create a Gradio interface
 interface = gr.Interface(