abhishekrs4
commited on
Commit
•
b456e16
1
Parent(s):
8ca797e
updated app script
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ num_classes = 5
|
|
20 |
image_height = 320
|
21 |
mean_arr = np.array([[0.485, 0.456, 0.406]])
|
22 |
std_arr = np.array([[0.229, 0.224, 0.225]])
|
23 |
-
cassava_model = ResNetImageClassififer(num_classes,
|
24 |
|
25 |
try:
|
26 |
logging.info(f"loading model from {file_model_local}")
|
|
|
20 |
image_height = 320
|
21 |
mean_arr = np.array([[0.485, 0.456, 0.406]])
|
22 |
std_arr = np.array([[0.229, 0.224, 0.225]])
|
23 |
+
cassava_model = ResNetImageClassififer(num_classes, pretrained=False)
|
24 |
|
25 |
try:
|
26 |
logging.info(f"loading model from {file_model_local}")
|