Update app.py

app.py

@@ -1,10 +1,20 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import gradio as gr
+import os
 
 # Load OpenHermes
 model_id = "BuddAi" if os.path.exists("BuddAi") else "CaptMetal/BuddAi"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
+try:
+    tokenizer = AutoTokenizer.from_pretrained(
+        model_id,
+        trust_remote_code=True  # Important for custom tokenizers
+    )
+    model = AutoModelForCausalLM.from_pretrained(
+        model_id,
+        device_map="auto"
+    )
+except Exception as e:
+    raise ValueError(f"Failed to load model: {str(e)}. Please verify all files exist in {model_id}")
 
 # Set chat template (OpenHermes-specific)
 tokenizer.chat_template = """{% for message in messages %}