Spaces:
Build error
Build error
printing a whole bunch of bullshit
Browse files
app.py
CHANGED
|
@@ -238,7 +238,11 @@ 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 |
|
| 243 |
print(model)
|
| 244 |
|
|
|
|
| 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 |
|