Spaces:
Build error
Build error
fixed renset bug
Browse files- predict_image.py +1 -1
predict_image.py
CHANGED
|
@@ -29,7 +29,7 @@ def load_model(model_name):
|
|
| 29 |
fname = download_weights(effb0_net_url)
|
| 30 |
print('loaded effnet')
|
| 31 |
elif model_name == 'res18':
|
| 32 |
-
model = ResnetModel
|
| 33 |
fname = download_weights(res18_net_url)
|
| 34 |
print('loaded resnet')
|
| 35 |
else:
|
|
|
|
| 29 |
fname = download_weights(effb0_net_url)
|
| 30 |
print('loaded effnet')
|
| 31 |
elif model_name == 'res18':
|
| 32 |
+
model = ResnetModel()
|
| 33 |
fname = download_weights(res18_net_url)
|
| 34 |
print('loaded resnet')
|
| 35 |
else:
|