Update app.py

app.py

@@ -1,11 +1,13 @@
 from keras.models import load_model
 import cv2
 import gradio as gr
+import os
 
 pox_model = load_model('fowl_pox_model.keras', compile=True)
-class_name={0:'Healthy',1:'Chicken have fowl pox',2:'Unkown'} 
-status={0:'Non Critical',1:'Critical',2:'N/A'}
-recommend={0:'No need medicine',1:'Panadol',2:'N/A'}
+class_name = {0: 'Healthy', 1: 'Chicken have fowl pox', 2: 'Unknown'}
+status = {0: 'Non Critical', 1: 'Critical', 2: 'N/A'}
+recommend = {0: 'No need medicine', 1: 'Panadol', 2: 'N/A'}
+
 def predict(img):
     # Resize the image to the required size for the model
     img_resized = cv2.resize(img, (256, 256))
@@ -47,14 +49,27 @@ def predict(img):
         cv2.rectangle(img_with_boxes, (min_x_scaled, min_y_scaled), (max_x_scaled, max_y_scaled), (0, 255, 0), 2)
     else:
         img_with_boxes = img
+    
+    # Save the output image
+    output_path = 'output_image.jpg'
+    cv2.imwrite(output_path, img_with_boxes)
 
-    return img_with_boxes, prediction_label, prediction_status, recommendation
+    return img_with_boxes, prediction_label, prediction_status, recommendation, output_path
 
 
-interface=gr.Interface(fn=predict,inputs='image',outputs=['image',gr.components.Textbox(label='Disease Name'),gr.components.Textbox(label='Disease status'),gr.components.Textbox(label='Disease medicine')],
-                      examples=[
-                          ['download (1).jpeg'],['download (2).jpeg'],['download (3).jpeg'],['images (1).jpeg'],['images (2).jpeg'],['images (3).jpeg']
-                      ])
+interface = gr.Interface(
+    fn=predict,
+    inputs='image',
+    outputs=[
+        'image',
+        gr.components.Textbox(label='Disease Name'),
+        gr.components.Textbox(label='Disease status'),
+        gr.components.Textbox(label='Disease medicine'),
+        gr.components.File(label='Download Image')
+    ],
+    examples=[
+        ['download (1).jpeg'], ['download (2).jpeg'], ['download (3).jpeg'],
+        ['images (1).jpeg'], ['images (2).jpeg'], ['images (3).jpeg']
+    ]
+)
 interface.launch(debug=True)
-
-