Spaces:

ambrosfitz
/

t5-history-MC

Runtime error

ambrosfitz commited on Oct 12, 2024

Commit

a5b5e3f

verified ·

1 Parent(s): 1d618ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import traceback
 error_message = ""
 # Load the model and tokenizer from Hugging Face
-model_name = "ambrosfitz/history-qa-t5-base"
 try:
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=False)

 error_message = ""
 # Load the model and tokenizer from Hugging Face
+model_name = "ambrosfitz/history-qa-t5-base-large"
 try:
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=False)