Update app.py

app.py

@@ -7,21 +7,26 @@ from transformers import (
 from threading import Thread
 
 # Configuration
-MODEL_NAME = "deepseek-ai/DeepSeek-R1"  # Verify exact model ID on Hugging Face Hub
+MODEL_NAME = "deepseek-ai/DeepSeek-R1"
 DEFAULT_MAX_NEW_TOKENS = 512
 
-# Load model and tokenizer
+# Load model and tokenizer WITH TRUSTED CODE
 try:
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+    tokenizer = AutoTokenizer.from_pretrained(
+        MODEL_NAME,
+        trust_remote_code=True  # <-- ADDED HERE
+    )
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_NAME,
         device_map="auto",
         torch_dtype="auto",
-        # load_in_4bit=True  # Uncomment for 4-bit quantization
+        trust_remote_code=True,  # <-- ADDED HERE
+        # load_in_4bit=True  # Uncomment for quantization
     )
 except Exception as e:
     raise gr.Error(f"Error loading model: {str(e)}")
 
+
 def generate_text(prompt, max_new_tokens=DEFAULT_MAX_NEW_TOKENS, temperature=0.7, top_p=0.9):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)