Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def analyse(img,plant_type):
|
|
16 |
# Load plant_label_disease.json
|
17 |
with open('data/plant_label_disease.json', 'r') as f:
|
18 |
plant_label_disease = json.load(f)
|
19 |
-
print(label_disease,label_disease[34])
|
20 |
HEIGHT = 256
|
21 |
WIDTH = 256
|
22 |
modelArchitecturePath ='model/model_architecture.h5'
|
@@ -30,13 +30,15 @@ def analyse(img,plant_type):
|
|
30 |
|
31 |
|
32 |
y_pred = dnn_model.predict(process_img)
|
|
|
33 |
print("y pred",y_pred)
|
34 |
indx = np.argmax(y_pred)
|
35 |
max_prob_indx = plant_label_disease[plant_type.lower()][0]
|
36 |
for disease in plant_label_disease[plant_type.lower()]:
|
37 |
-
|
38 |
-
if y_pred[0][disease]>y_pred[0][max_prob_indx]:
|
39 |
max_prob_indx = disease
|
|
|
|
|
40 |
print(label_disease[indx])
|
41 |
print(y_pred[0][indx])
|
42 |
print(label_disease[max_prob_indx])
|
|
|
16 |
# Load plant_label_disease.json
|
17 |
with open('data/plant_label_disease.json', 'r') as f:
|
18 |
plant_label_disease = json.load(f)
|
19 |
+
print(label_disease,label_disease["34"])
|
20 |
HEIGHT = 256
|
21 |
WIDTH = 256
|
22 |
modelArchitecturePath ='model/model_architecture.h5'
|
|
|
30 |
|
31 |
|
32 |
y_pred = dnn_model.predict(process_img)
|
33 |
+
y_pred = y_pred[0]
|
34 |
print("y pred",y_pred)
|
35 |
indx = np.argmax(y_pred)
|
36 |
max_prob_indx = plant_label_disease[plant_type.lower()][0]
|
37 |
for disease in plant_label_disease[plant_type.lower()]:
|
38 |
+
if y_pred[disease] > y_pred[max_prob_indx]:
|
|
|
39 |
max_prob_indx = disease
|
40 |
+
indx=str(indx)
|
41 |
+
max_prob_indx=str(max_prob_indx)
|
42 |
print(label_disease[indx])
|
43 |
print(y_pred[0][indx])
|
44 |
print(label_disease[max_prob_indx])
|