Update app.py

app.py

@@ -52,13 +52,33 @@ def predict_disease(image):
     # Get the predicted label
     predicted_label = class_names[predicted_class.item()]
     
-    return predicted_label
+    # Create a styled HTML output
+    output_message = f"""
+    <div style='font-size: 18px; color: #4CAF50; font-weight: bold;'>
+        Detected Disease: <span style='color: #FF5722;'>{predicted_label}</span>
+    </div>
+    """
+    
+    if predicted_label != "Healthy":
+        output_message += f"""
+        <p style='font-size: 16px; color: #757575;'>
+            This indicates the presence of <strong>{predicted_label}</strong>. Please take immediate action to prevent further spread.
+        </p>
+        """
+    else:
+        output_message += f"""
+        <p style='font-size: 16px; color: #757575;'>
+            The sugarcane crop is <strong>healthy</strong>. Keep monitoring for potential risks.
+        </p>
+        """
+    
+    return output_message
 
 # Create Gradio interface
 inputs = gr.Image(type="pil")
-outputs = gr.Text()
+outputs = gr.HTML()  # Use HTML output for styled text
 
-EXAMPLES = ["img1.jpeg","redrot2.jpg","rust1.jpg","healthy2.jpeg"]
+EXAMPLES = ["img1.jpeg", "redrot2.jpg", "rust1.jpg", "healthy2.jpeg"]
 
 demo_app = gr.Interface(
     fn=predict_disease,
@@ -70,7 +90,4 @@ demo_app = gr.Interface(
     theme="huggingface"
 )
 
-demo_app.launch(debug=True)
-
-
-
+demo_app.launch(debug=True)