Update app.py

app.py

@@ -10,8 +10,8 @@ os.system("pip install git+https://github.com/shumingma/transformers.git")
 torch._dynamo.config.suppress_errors = True
 
 model_id = "microsoft/bitnet-b1.58-2B-4T"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32, device_map={"": "cpu"}).to("cpu")
+tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32, device_map={"": "cpu"}, trust_remote_code=True)
 model.to("cpu")
 
 @spaces.GPU(duration=15)