p1atdev commited on
Commit
8860ed6
·
1 Parent(s): 8e787dd

chore: set token

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import numpy as np
2
  import torch
3
 
@@ -5,7 +8,6 @@ from transformers import (
5
  AutoProcessor,
6
  )
7
 
8
- from PIL import Image
9
  import gradio as gr
10
 
11
  from modeling_siglip import SiglipForImageClassification
@@ -15,6 +17,7 @@ PROCESSOR_NAME = "google/siglip-so400m-patch14-384"
15
 
16
  model = SiglipForImageClassification.from_pretrained(
17
  MODEL_NAME,
 
18
  )
19
  # model = torch.compile(model)
20
  processor = AutoProcessor.from_pretrained(PROCESSOR_NAME)
 
1
+ import os
2
+ from PIL import Image
3
+
4
  import numpy as np
5
  import torch
6
 
 
8
  AutoProcessor,
9
  )
10
 
 
11
  import gradio as gr
12
 
13
  from modeling_siglip import SiglipForImageClassification
 
17
 
18
  model = SiglipForImageClassification.from_pretrained(
19
  MODEL_NAME,
20
+ token=os.environ["HF_READ_TOKEN"],
21
  )
22
  # model = torch.compile(model)
23
  processor = AutoProcessor.from_pretrained(PROCESSOR_NAME)