Spaces:
Build error
Build error
getting rid of some statements
Browse files
app.py
CHANGED
|
@@ -296,10 +296,10 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
|
|
| 296 |
# print(model_url)
|
| 297 |
|
| 298 |
loaded = torch.load(model_url, map_location='cpu', )
|
| 299 |
-
print(loaded)
|
| 300 |
|
| 301 |
model.load_state_dict(loaded)
|
| 302 |
-
print(model)
|
| 303 |
|
| 304 |
# model = EfficientNetPreTrained(config)
|
| 305 |
# config.register_for_auto_class()
|
|
|
|
| 296 |
# print(model_url)
|
| 297 |
|
| 298 |
loaded = torch.load(model_url, map_location='cpu', )
|
| 299 |
+
# print(loaded)
|
| 300 |
|
| 301 |
model.load_state_dict(loaded)
|
| 302 |
+
# print(model)
|
| 303 |
|
| 304 |
# model = EfficientNetPreTrained(config)
|
| 305 |
# config.register_for_auto_class()
|