updated

app.py

@@ -6,6 +6,10 @@ from PIL import Image
 
 from torchvision.transforms.functional import InterpolationMode
 # Define the path to your model
+import os
+from huggingface_hub import login
+hf_token = os.environ.get('hf_token', None)
+
 path = 'h2oai/h2o-mississippi-2b'
 
 # image preprocesing
@@ -156,13 +160,15 @@ model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
-    trust_remote_code=True
+    trust_remote_code=True,
+    use_auth_token=hf_token
 ).eval().cuda()
 
 tokenizer = AutoTokenizer.from_pretrained(
     path, 
     trust_remote_code=True, 
-    use_fast=False
+    use_fast=False,
+    use_auth_token=hf_token
 )
 tokenizer.pad_token = tokenizer.unk_token
 tokenizer.eos_token = "<|end|>"