conan1024hao commited on
Commit
ff4fade
1 Parent(s): 3cd8568

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("./mt5")
6
- model = AutoModelForSeq2SeqLM.from_pretrained("./mt5")
7
  model.resize_token_embeddings(len(tokenizer))
8
  model.eval()
9
 
 
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
 
5
+ tokenizer = AutoTokenizer.from_pretrained("mt5")
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("mt5")
7
  model.resize_token_embeddings(len(tokenizer))
8
  model.eval()
9