printing url
Browse files
app.py
CHANGED
@@ -238,12 +238,12 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
|
|
238 |
depth_mult=hparams.depth_mult,)
|
239 |
|
240 |
model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
|
|
|
241 |
|
242 |
loaded = torch.load(model_url, map_location='cpu')
|
243 |
print(loaded)
|
244 |
|
245 |
model.load_state_dict(loaded)
|
246 |
-
|
247 |
print(model)
|
248 |
|
249 |
# model = EfficientNetPreTrained(config)
|
|
|
238 |
depth_mult=hparams.depth_mult,)
|
239 |
|
240 |
model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
|
241 |
+
print(model_url)
|
242 |
|
243 |
loaded = torch.load(model_url, map_location='cpu')
|
244 |
print(loaded)
|
245 |
|
246 |
model.load_state_dict(loaded)
|
|
|
247 |
print(model)
|
248 |
|
249 |
# model = EfficientNetPreTrained(config)
|