Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -32,7 +32,7 @@ class EndpointHandler():
|
|
| 32 |
self.blip_encoder = BLIP_Base(
|
| 33 |
image_size=image_size,
|
| 34 |
vit='base',
|
| 35 |
-
med_config='configs/med_config.json'
|
| 36 |
)
|
| 37 |
load_checkpoint(self.blip_encoder, self.blip_encoder_path)
|
| 38 |
self.blip_encoder.eval()
|
|
|
|
| 32 |
self.blip_encoder = BLIP_Base(
|
| 33 |
image_size=image_size,
|
| 34 |
vit='base',
|
| 35 |
+
med_config='/repository/configs/med_config.json'
|
| 36 |
)
|
| 37 |
load_checkpoint(self.blip_encoder, self.blip_encoder_path)
|
| 38 |
self.blip_encoder.eval()
|