juanesbch commited on
Commit
e2da54a
β€’
1 Parent(s): cc80207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
 
2
  import transformers as trf
3
 
4
  # Load the summarization model
5
- summarization_model_path = '"models/ieuniversity/summarization_model_translator"'
6
- summarization_tokenizer = trf.AutoTokenizer.from_pretrained(summarization_model_path)
7
  summarization_model = trf.AutoModelForSeq2SeqLM.from_pretrained(summarization_model_path)
8
 
9
  # Load the translation model
 
1
  import gradio as gr
2
+ import torch
3
  import transformers as trf
4
 
5
  # Load the summarization model
6
+ summarization_model_path = 'ieuniversity/summarization_model_translator'
7
+ summarization_tokenizer = trf.AutoTokenizer.from_pretrained("it5/it5-base-news-summarization")
8
  summarization_model = trf.AutoModelForSeq2SeqLM.from_pretrained(summarization_model_path)
9
 
10
  # Load the translation model