Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
+import torch
 import transformers as trf
 
 # Load the summarization model
-summarization_model_path = '"models/ieuniversity/summarization_model_translator"'
-summarization_tokenizer = trf.AutoTokenizer.from_pretrained(summarization_model_path)
+summarization_model_path = 'ieuniversity/summarization_model_translator'
+summarization_tokenizer = trf.AutoTokenizer.from_pretrained("it5/it5-base-news-summarization")
 summarization_model = trf.AutoModelForSeq2SeqLM.from_pretrained(summarization_model_path)
 
 # Load the translation model