jjuarez commited on
Commit
b7db3d0
·
1 Parent(s): 3e76600

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
 
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("jjuarez/translation_model")
6
- model = AutoModelForSeq2SeqLM.from_pretrained("jjuarez/translation_model")
7
 
8
  # Define the translation function
9
  def translate(text):
 
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
+ tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_translation")
5
 
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
 
7
 
8
  # Define the translation function
9
  def translate(text):