Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = '
|
6 |
-
summarization_tokenizer = trf.AutoTokenizer.from_pretrained(
|
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
|