oiisa commited on
Commit
2d57502
·
verified ·
1 Parent(s): f7514ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def load_model():
69
  raise FileNotFoundError(f"Model directory not found at {model_path}")
70
 
71
  try:
72
- tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small")
73
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
74
  model.eval()
75
  return model, tokenizer
 
69
  raise FileNotFoundError(f"Model directory not found at {model_path}")
70
 
71
  try:
72
+ tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-base")
73
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
74
  model.eval()
75
  return model, tokenizer