Update app.py

app.py

@@ -13,10 +13,13 @@ print(f"Loading tokenizer and model from {MODEL_ID} on {DEVICE}")
 
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=True)
-# If the model was pushed with custom config (like trusting remote code), we handle gracefully
-model = AutoModelForCausalLM.from_pretrained(MODEL_ID, trust_remote_code=True)
+
+# Safe loader
+try:
+    model = AutoModelForCausalLM.from_pretrained(MODEL_ID, trust_remote_code=True)
 except Exception:
-model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
+    model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
+
 
 
 model.to(DEVICE)