Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
 from transformers import pipeline
-from transformers import AutoTokenizer, AutoModelForCasualLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Wonder-Griffin/TraXL")
+tokenizer = AutoTokenizer.from_pretrained("Wonder-Griffin/TraXLMistral")
 
-pipe = pipeline("text-generation", model="Wonder-Griffin/TraXL")
+pipe = pipeline("text-generation", model="Wonder-Griffin/TraXLMistral")
 
 # Assuming `model_path` is the Hugging Face model hub path or a local directory
-model_path = "Wonder-Griffin/TraXL"  # Define this as needed
+model_path = "Wonder-Griffin/TraXLMistral"  # Define this as needed
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCasualLM.from_pretrained(
+model = AutoModelForCausalLM.from_pretrained(
     model_path,
     device_map="auto",
     torch_dtype='auto'