Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 from diffusers import DiffusionPipeline
 from datasets import load_dataset
 
-headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generate-headline")
+headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generate-headline", tokenizer="t5-base")
 ar_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en")
 en_to_ar_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar")