Update app.py

app.py

@@ -96,7 +96,7 @@ def classify_image(inp):
     modulator = (modulator - modulator.min()) / (modulator.max() - modulator.min())
     cam3 = show_cam_on_image(img_d, modulator, use_rgb=True)
 
-    return {labels[i]: float(prediction[i]) for i in range(1000)}, Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3)
+    return {labels[i]: float(prediction[i]) for i in range(1000)}, Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3), Image.fromarray(img_d)
 
 
 image = gr.inputs.Image()
@@ -119,7 +119,10 @@ gr.Interface(
         label="Modulator at layer 6"),     
         gr.outputs.Image(
         type="pil",
-        label="Modulator at layer 3"),                        
+        label="Modulator at layer 3"),     
+        gr.outputs.Image(
+        type="pil",
+        label="Cropped Input"),                                
     ],
     examples=[["./donut.png"], ["./horses.png"], ["./pencil.png"]],
 ).launch()