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()
|