project21 commited on
Commit
4f10806
·
verified ·
1 Parent(s): 77d4e22

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +209 -0
app.py ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from tensorflow.keras.utils import img_to_array,load_img
3
+ from keras.models import load_model
4
+ import numpy as np
5
+
6
+ # Load the pre-trained model from the local path
7
+ model_path = 'corn.h5'
8
+ model = load_model(model_path) # Load the model here
9
+
10
+ def predict_disease(image_file, model, all_labels):
11
+
12
+ try:
13
+ # Load and preprocess the image
14
+ img = load_img(image_file, target_size=(224, 224)) # Use load_img from tensorflow.keras.utils
15
+ img_array = img_to_array(img)
16
+ img_array = np.expand_dims(img_array, axis=0) # Add batch dimension
17
+ img_array = img_array / 255.0 # Normalize the image
18
+
19
+ # Predict the class
20
+ predictions = model.predict(img_array) # Use the loaded model here
21
+ predicted_class = np.argmax(predictions[0])
22
+
23
+ # Get the predicted class label
24
+ predicted_label = all_labels[predicted_class]
25
+
26
+ # Print the predicted label to the console
27
+
28
+ if predicted_label=='Corn Healthy':
29
+ predicted_label = predicted_label = """<h3 align="center">Corn Healthy</h3><br><br>
30
+ <center>No need use Pesticides</center>"""
31
+ elif predicted_label=='Corn Northern Leaf Blight':
32
+ predicted_label = """
33
+ <style>
34
+ li{
35
+ font-size: 15px;
36
+ margin-left: 90px;
37
+ margin-top: 15px;
38
+ margin-bottom: 15px;
39
+ }
40
+ h4{
41
+ font-size: 17px;
42
+ margin-top: 15px;
43
+ }
44
+ h4:hover{
45
+ cursor: pointer;
46
+ }
47
+
48
+ h3:hover{
49
+ cursor: pointer;
50
+ color: blue;
51
+ transform: scale(1.3);
52
+ }
53
+ .note{
54
+ text-align: center;
55
+ font-size: 16px;
56
+ }
57
+ p{
58
+ font-size: 13px;
59
+ text-align: center;
60
+ }
61
+
62
+ </style>
63
+ <h3><center><b>Corn Northern Leaf Blight</b></center></h3>
64
+ <h4>PESTICIDES TO BE USED:</h4>
65
+ <ul>
66
+ <li>1. Chlorothalonil (Daconil)</li>
67
+ <li>2. Mancozeb (Dithane)</li>
68
+ <li>3. Copper oxychloride (Kocide)</li>
69
+ <li>4. Azoxystrobin (Heritage)</li>
70
+ <li>5. Pyraclostrobin (Cabrio)</li>
71
+
72
+
73
+ </ul>
74
+ <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
75
+ <p>Be sure to follow local regulations and guidelines for application</p>
76
+
77
+
78
+ """
79
+ elif predicted_label=='Corn Gray leaf spot':
80
+ predicted_label = """
81
+ <style>
82
+ li{
83
+ font-size: 15px;
84
+ margin-left: 90px;
85
+ margin-top: 15px;
86
+ margin-bottom: 15px;
87
+ }
88
+ h4{
89
+ font-size: 17px;
90
+ margin-top: 15px;
91
+ }
92
+ h4:hover{
93
+ cursor: pointer;
94
+ }
95
+
96
+ h3:hover{
97
+ cursor: pointer;
98
+ color: blue;
99
+ transform: scale(1.3);
100
+ }
101
+ .note{
102
+ text-align: center;
103
+ font-size: 16px;
104
+ }
105
+ p{
106
+ font-size: 13px;
107
+ text-align: center;
108
+ }
109
+
110
+ </style>
111
+ <h3><center><b>Corn Gray leaf spot</b></center></h3>
112
+ <h4>PESTICIDES TO BE USED:</h4>
113
+ <ul>
114
+ <li>1. Copper oxychloride (Kocide)</li>
115
+ <li>2. Mancozeb(Dithane)</li>
116
+ <li>3. Azoxystrobin</li>
117
+ <li>4. Chlorothalonil</li>
118
+
119
+
120
+ </ul>
121
+ <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
122
+ <p>Be sure to follow local regulations and guidelines for application</p>
123
+
124
+
125
+ """
126
+ elif predicted_label=='Corn Common Rust':
127
+ predicted_label = """
128
+ <style>
129
+ li{
130
+ font-size: 15px;
131
+ margin-left: 90px;
132
+ margin-top: 15px;
133
+ margin-bottom: 15px;
134
+ }
135
+ h4{
136
+ font-size: 17px;
137
+ margin-top: 15px;
138
+ }
139
+ h4:hover{
140
+ cursor: pointer;
141
+ }
142
+
143
+ h3:hover{
144
+ cursor: pointer;
145
+ color: blue;
146
+ transform: scale(1.3);
147
+ }
148
+ .note{
149
+ text-align: center;
150
+ font-size: 16px;
151
+ }
152
+ p{
153
+ font-size: 13px;
154
+ text-align: center;
155
+ }
156
+
157
+ </style>
158
+ <h3><center><b>Corn Common Rust</b></center></h3>
159
+ <h4>PESTICIDES TO BE USED:</h4>
160
+ <ul>
161
+ <li>1. Copper oxychloride (Kocide)</li>
162
+ <li>2. Mancozeb(Dithane)</li>
163
+ <li>3. Azoxystrobin</li>
164
+ <li>4. Chlorothalonil</li>
165
+
166
+
167
+ </ul>
168
+ <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
169
+ <p>Be sure to follow local regulations and guidelines for application</p>
170
+
171
+
172
+ """
173
+
174
+
175
+
176
+ else:
177
+ predicted_label = """<h3 align="center">Choose Correct image</h3><br><br>
178
+ """
179
+
180
+ return predicted_label
181
+
182
+
183
+ except Exception as e:
184
+ print(f"Error: {e}")
185
+ return None
186
+
187
+ # List of class labels
188
+ all_labels = [
189
+ 'Corn Northern Leaf Blight',
190
+ 'Corn Healthy',
191
+ 'Corn Gray leaf spot',
192
+ 'Corn Common Rust'
193
+ ]
194
+
195
+ # Define the Gradio interface
196
+ def gradio_predict(image_file):
197
+ return predict_disease(image_file, model, all_labels) # Pass the model to the function
198
+
199
+ # Create a Gradio interface
200
+ gr_interface = gr.Interface(
201
+ fn=gradio_predict, # Function to call for predictions
202
+ inputs=gr.Image(type="filepath"), # Upload image as file path
203
+ outputs="html", # Output will be the class label as text
204
+ title="Corn Disease Predictor",
205
+ description="Upload an image of a plant to predict the disease.",
206
+ )
207
+
208
+ # Launch the Gradio app
209
+ gr_interface.launch(share=True)