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), Image.fromarray(img_d)
+    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(np.uint8(255 * img_d))
 
 
 image = gr.inputs.Image()