Update app.py

app.py

@@ -24,12 +24,14 @@ quantization_config = BitsAndBytesConfig(
     bnb_4bit_use_double_quant=True,
     bnb_4bit_quant_type="nf4")
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID1)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID1, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID1,
     torch_dtype=torch.bfloat16,
     device_map="auto",
-    quantization_config=quantization_config)
+    quantization_config=quantization_config,
+    trust_remote_code=True
+    )
 
 # Chatbot tab function
 @spaces.GPU()