Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| # Import tensorflow here | |
| import tensorflow as tf | |
| from tensorflow.keras.utils import img_to_array,load_img | |
| from tensorflow.keras.models import load_model # Use tensorflow.keras.models | |
| import numpy as np | |
| # Load the pre-trained model from the local path | |
| model_path = 'Mango.h5' | |
| # Define custom objects to handle potential incompatibilities | |
| custom_objects = {'DepthwiseConv2D': tf.keras.layers.DepthwiseConv2D} | |
| # Load the model with custom_objects | |
| model = load_model(model_path, custom_objects=custom_objects) # Load the model here | |
| def predict_disease(image_file, model, all_labels): | |
| try: | |
| # Load and preprocess the image | |
| img = load_img(image_file, target_size=(224, 224)) # Use load_img from tensorflow.keras.utils | |
| img_array = img_to_array(img) | |
| img_array = np.expand_dims(img_array, axis=0) # Add batch dimension | |
| img_array = img_array / 255.0 # Normalize the image | |
| # Predict the class | |
| predictions = model.predict(img_array) # Use the loaded model here | |
| predicted_class = np.argmax(predictions[0]) | |
| # Get the predicted class label | |
| predicted_label = all_labels[predicted_class] | |
| # Print the predicted label to the console | |
| if predicted_label=='Mango Anthracrose': | |
| predicted_label = """<style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Anthracrose</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Mancozeb</li> | |
| <li>2. Azoxystrobin</li> | |
| <li>3. carbendazim</li> | |
| <li>4. Propiconazole</li> | |
| <li>5. Thiophanate-methyl</li> | |
| <li>6. Copper Sulfate</li> | |
| </ul><br> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Bacterial Canker': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Bacterial Canker</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Copper Hydroxide</li> | |
| <li>2. Copper Oxychloride</li> | |
| <li>3. Streptomycin</li> | |
| <li>4. oxytetracycline</li> | |
| <li>5. Neem oil</li> | |
| <li>6. Garlic oil</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Cutting Weevil': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Cutting Weevil</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Imidacloprid</li> | |
| <li>2. Thiamethoxam</li> | |
| <li>3. Chlorpyrifos</li> | |
| <li>4. Lambda-cyhalothrin</li> | |
| <li>5. Fipronil</li> | |
| <li>6. Neem oil</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Die Back': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Die Back</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Carbendazim</li> | |
| <li>2. Mancozeb</li> | |
| <li>3. Azoxystrobin</li> | |
| <li>4. Triazole</li> | |
| <li>5. Potassium bicarbonate</li> | |
| <li>6. Sodium bicarbonate</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Gall Midge': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Gall Midge</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Imidacloprid</li> | |
| <li>2. Thiamethoxam</li> | |
| <li>3. Chlorpyrifos</li> | |
| <li>4. Lambda-cyhalothrin</li> | |
| <li>5. Spinosad</li> | |
| <li>6. Pyrethrin</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Powdery Mildew': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Powdery Mildew</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Sulfur</li> | |
| <li>2. Bicarbonates</li> | |
| <li>3. Myclobutanil</li> | |
| <li>4. Triadimefon</li> | |
| <li>5. Propiconazole</li> | |
| <li>6. Azoxystrobin</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| elif predicted_label=='Mango Sooty Mould': | |
| predicted_label = """ | |
| <style> | |
| li{ | |
| font-size: 15px; | |
| margin-left: 90px; | |
| margin-top: 15px; | |
| margin-bottom: 15px; | |
| } | |
| h4{ | |
| font-size: 17px; | |
| margin-top: 15px; | |
| } | |
| h4:hover{ | |
| cursor: pointer; | |
| } | |
| h3:hover{ | |
| cursor: pointer; | |
| color: blue; | |
| transform: scale(1.3); | |
| } | |
| .note{ | |
| text-align: center; | |
| font-size: 16px; | |
| } | |
| p{ | |
| font-size: 13px; | |
| text-align: center; | |
| } | |
| </style> | |
| <h3><center><b>Mango Sooty Mould</b></center></h3> | |
| <h4>PESTICIDES TO BE USED:</h4> | |
| <ul> | |
| <li>1. Imidacloprid (Neonicotinoid)</li> | |
| <li>2. Thiamethoxam (Neonicotinoid)</li> | |
| <li>3. Bifenthrin (Pyrethroid)</li> | |
| <li>4. Lambda-cyhalothrin (Pyrethroid)</li> | |
| <li>5. Insecticidal soap</li> | |
| <li>6. Horticultural oil</li> | |
| </ul> | |
| <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p> | |
| <p>Be sure to follow local regulations and guidelines for application</p> | |
| """ | |
| else: | |
| predicted_label = """<h3 align="center">Mango Healthy</h3><br><br> | |
| <center>No need use Pesticides</center>""" | |
| return predicted_label | |
| except Exception as e: | |
| print(f"Error: {e}") | |
| return None | |
| # List of class labels | |
| all_labels = [ | |
| 'Mango Anthracrose', | |
| 'Mango Bacterial Canker', | |
| 'Mango Cutting Weevil', | |
| 'Mango Die Back', | |
| 'Mango Gall Midge', | |
| 'Mango Healthy', | |
| 'Mango Powdery Mildew', | |
| 'Mango Sooty Mould' | |
| ] | |
| # Define the Gradio interface | |
| def gradio_predict(image_file): | |
| return predict_disease(image_file, model, all_labels) # Pass the model to the function | |
| # Create a Gradio interface | |
| gr_interface = gr.Interface( | |
| fn=gradio_predict, # Function to call for predictions | |
| inputs=gr.Image(type="filepath"), # Upload image as file path | |
| outputs="html", # Output will be the class label as text | |
| title="""<center>Mango Disease Predictor</center>""", | |
| description="Upload an image of a plant to predict the disease.", | |
| ) | |
| # Launch the Gradio app | |
| gr_interface.launch(share=True) |