Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 from transformers import AutoModel, AutoProcessor
 import torch
 import requests
-from transformers import pipeline
+from PIL import Image
+from io import BytesIO
 
 # Import a specific model
 model_name = "facebook/bart-large-cnn"
@@ -13,7 +14,7 @@ summarizer = pipeline("summarization", model=model_name)
 
 # Define a function that takes a text and returns a summary
 def summarize(input_text):
-  summary = summarizer(input_text, max_length=150, min_length=40, do_sample=False)
+  summary = summarizer(input_text, max_length=100, min_length=10, do_sample=False)
   return summary[0]["summary_text"]
 
 # Create a Gradio interface