Update app.py

app.py

@@ -95,7 +95,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 Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3), {labels[i]: float(prediction[i]) for i in range(1000)}
+    return {labels[i]: float(prediction[i]) for i in range(1000)}, Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3)
 
 
 image = gr.inputs.Image()
@@ -106,6 +106,7 @@ gr.Interface(
     fn=classify_image,
     inputs=image,
     outputs=[
+        label,
         gr.outputs.Image(
         type="pil",
         label="Modulator at layer 12"),     
@@ -118,7 +119,6 @@ gr.Interface(
         gr.outputs.Image(
         type="pil",
         label="Modulator at layer 3"),                        
-        label,
     ],
     examples=[["./donut.png"], ["./horses.png"], ["./pencil.png"]],
 ).launch()