samwell commited on
Commit
53eb068
·
verified ·
1 Parent(s): 258c09c

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -77,7 +77,9 @@ class LaborViewMedSigLIP(nn.Module):
77
  super().__init__()
78
  self.config = config
79
  from transformers import AutoModel
80
- self.encoder = AutoModel.from_pretrained(config.encoder_pretrained, trust_remote_code=True)
 
 
81
  if hasattr(self.encoder, 'vision_model'):
82
  self.vision_encoder = self.encoder.vision_model
83
  else:
 
77
  super().__init__()
78
  self.config = config
79
  from transformers import AutoModel
80
+ import os
81
+ token = os.environ.get("HF_TOKEN")
82
+ self.encoder = AutoModel.from_pretrained(config.encoder_pretrained, trust_remote_code=True, token=token)
83
  if hasattr(self.encoder, 'vision_model'):
84
  self.vision_encoder = self.encoder.vision_model
85
  else: