Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from transformers import pipeline
 import torch
 from diffusers import DiffusionPipeline
+from transformers import pipeline, AutoTokenizer
+
 
 headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generate-headline")
 ar_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en")