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:
|