Update app.py

app.py

@@ -1,12 +1,18 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaTokenizer
 import torch
 
 # Load the model and tokenizer
 model_name = "cognitivecomputations/TinyDolphin-2.8-1.1b"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+# Try to load the tokenizer, with a fallback option
+try:
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+except ValueError:
+    print("Failed to load AutoTokenizer. Falling back to LlamaTokenizer.")
+    tokenizer = LlamaTokenizer.from_pretrained(model_name)
+
 # Move model to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)