Update app.py

app.py

@@ -13,11 +13,11 @@ checkpoint = "vsrinivas/FalconLite2"
 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto", 
     offload_folder="off_load", 
-    # trust_remote_code=True,
+    trust_remote_code=True,
     # torch_dtype="auto",
 )
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, 
-                                          # trust_remote_code=True,
+                                          trust_remote_code=True,
                                           torch_dtype="auto",
                                          )
 
@@ -32,7 +32,7 @@ pipeline = transformers.pipeline(
     tokenizer=tokenizer,
     # use_safetensors=True, 
     # torch_dtype=torch.bfloat16,
-    # trust_remote_code=True,
+    trust_remote_code=True,
     device_map="auto",
     offload_folder="off_load",
     offload_state_dict = True,