Spaces:
Runtime error
Runtime error
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -77,7 +77,9 @@ class LaborViewMedSigLIP(nn.Module):
|
|
| 77 |
super().__init__()
|
| 78 |
self.config = config
|
| 79 |
from transformers import AutoModel
|
| 80 |
-
|
|
|
|
|
|
|
| 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:
|