Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def classify_image(inp):
|
|
95 |
modulator = (modulator - modulator.min()) / (modulator.max() - modulator.min())
|
96 |
cam3 = show_cam_on_image(img_d, modulator, use_rgb=True)
|
97 |
|
98 |
-
return Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3)
|
99 |
|
100 |
|
101 |
image = gr.inputs.Image()
|
@@ -106,6 +106,7 @@ gr.Interface(
|
|
106 |
fn=classify_image,
|
107 |
inputs=image,
|
108 |
outputs=[
|
|
|
109 |
gr.outputs.Image(
|
110 |
type="pil",
|
111 |
label="Modulator at layer 12"),
|
@@ -118,7 +119,6 @@ gr.Interface(
|
|
118 |
gr.outputs.Image(
|
119 |
type="pil",
|
120 |
label="Modulator at layer 3"),
|
121 |
-
label,
|
122 |
],
|
123 |
examples=[["./donut.png"], ["./horses.png"], ["./pencil.png"]],
|
124 |
).launch()
|
|
|
95 |
modulator = (modulator - modulator.min()) / (modulator.max() - modulator.min())
|
96 |
cam3 = show_cam_on_image(img_d, modulator, use_rgb=True)
|
97 |
|
98 |
+
return {labels[i]: float(prediction[i]) for i in range(1000)}, Image.fromarray(cam0), Image.fromarray(cam1), Image.fromarray(cam2), Image.fromarray(cam3)
|
99 |
|
100 |
|
101 |
image = gr.inputs.Image()
|
|
|
106 |
fn=classify_image,
|
107 |
inputs=image,
|
108 |
outputs=[
|
109 |
+
label,
|
110 |
gr.outputs.Image(
|
111 |
type="pil",
|
112 |
label="Modulator at layer 12"),
|
|
|
119 |
gr.outputs.Image(
|
120 |
type="pil",
|
121 |
label="Modulator at layer 3"),
|
|
|
122 |
],
|
123 |
examples=[["./donut.png"], ["./horses.png"], ["./pencil.png"]],
|
124 |
).launch()
|