adds auth_token as env var

app.ipynb

@@ -8,6 +8,7 @@
    "source": [
     "import gradio as gr\n",
     "import numpy as np\n",
+    "from os import environ\n",
     "from PIL import Image as PImage\n",
     "from torchvision import transforms as T\n",
     "from transformers import MaskFormerForInstanceSegmentation, MaskFormerImageProcessor"
@@ -40,7 +41,8 @@
     "    reduce_labels=False\n",
     ")\n",
     "\n",
-    "model = MaskFormerForInstanceSegmentation.from_pretrained(model_id)\n",
+    "hf_token = environ.get('HFTOKEN') or True\n",
+    "model = MaskFormerForInstanceSegmentation.from_pretrained(model_id, use_auth_token=hf_token)\n",
     "\n",
     "test_transform = T.Compose([\n",
     "    T.ToTensor(),\n",

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import numpy as np
+from os import environ
 from torchvision import transforms as T
 from transformers import MaskFormerForInstanceSegmentation, MaskFormerImageProcessor
 
@@ -23,7 +24,8 @@ preprocessor = MaskFormerImageProcessor(
     reduce_labels=False
 )
 
-model = MaskFormerForInstanceSegmentation.from_pretrained(model_id)
+hf_token = environ.get('HFTOKEN')
+model = MaskFormerForInstanceSegmentation.from_pretrained(model_id, use_auth_token=hf_token)
 
 test_transform = T.Compose([
     T.ToTensor(),