Update app.py

app.py

@@ -1,11 +1,15 @@
 import gradio as gr
 import torch
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM, T5Tokenizer
 
 # Load the model and tokenizer from Hugging Face
 model_name = "ambrosfitz/history-qa-t5-base"
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+try:
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=False)
+except Exception as e:
+    print(f"Error loading model or tokenizer: {e}")
+    raise
 
 # Set device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")