Update app.py

app.py

@@ -2,11 +2,15 @@ import gradio as gr
 import numpy as np
 import torch
 from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
+import os
+access_token = os.getenv('HF_TOKEN')
+from huggingface_hub import login
+login(token = access_token)
 
 # Load the model from Hugging Face
 model_name = "gdurkin/cdl_mask2former_hi_res_v3"
-processor = Mask2FormerImageProcessor.from_pretrained(model_name)
-model = Mask2FormerForUniversalSegmentation.from_pretrained(model_name)
+processor = Mask2FormerImageProcessor.from_pretrained(model_name,token = access_token)
+model = Mask2FormerForUniversalSegmentation.from_pretrained(model_name,token = access_token)
 device = torch.device('cpu')
 
 # Define the inference function