Spaces:
Runtime error
Runtime error
Commit
·
42ecc5d
1
Parent(s):
73c6f3e
Update app.py
Browse files
app.py
CHANGED
@@ -237,7 +237,7 @@ def server(input: Inputs, output: Outputs, session: Session):
|
|
237 |
plt.tight_layout()
|
238 |
return fig
|
239 |
|
240 |
-
fig, axes = plt.subplots(1, 4,
|
241 |
|
242 |
# Extract the image data
|
243 |
#image_data = image.cpu().detach().numpy()
|
@@ -246,9 +246,9 @@ def server(input: Inputs, output: Outputs, session: Session):
|
|
246 |
axes[0].imshow(image)
|
247 |
axes[0].set_title("Image")
|
248 |
|
249 |
-
#
|
250 |
-
|
251 |
-
|
252 |
|
253 |
# # Plot the prediction image on the right
|
254 |
# axes[2].imshow(prediction)
|
|
|
237 |
plt.tight_layout()
|
238 |
return fig
|
239 |
|
240 |
+
fig, axes = plt.subplots(1, 4, , figsize=(15, 5))
|
241 |
|
242 |
# Extract the image data
|
243 |
#image_data = image.cpu().detach().numpy()
|
|
|
246 |
axes[0].imshow(image)
|
247 |
axes[0].set_title("Image")
|
248 |
|
249 |
+
# Plot the probability map on the right
|
250 |
+
axes[1].imshow(prob)
|
251 |
+
axes[1].set_title("Probability Map")
|
252 |
|
253 |
# # Plot the prediction image on the right
|
254 |
# axes[2].imshow(prediction)
|