Spaces:
Build error
Build error
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)
|