bhanusAI commited on
Commit
7bc8f9c
·
verified ·
1 Parent(s): b3ccf32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- print(disease,y_pred[0][disease],max_prob_indx,y_pred[0][max_prob_indx])
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])