Update app.py

app.py

@@ -2,18 +2,18 @@ import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
 
 # Load the model and tokenizer
-model_name = 'IMISLab/GreekWiki-umt5-base'
+model_name = 'IMISLab/GreekT5-umt5-small-greeksum'
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Set up the summarizer pipeline
 summarizer = pipeline(
     'summarization',
-    model=model,
-    tokenizer=tokenizer,
-    device=-1,  # -1 for CPU; set to 0 for GPU if available
-    max_new_tokens=220,
-    truncation=True
+    device = 'cpu',
+    model = model,
+    tokenizer = tokenizer,
+    max_new_tokens = 128,
+    truncation = True
 )
 
 # Define the summarization function