aagoluoglu commited on
Commit
42ecc5d
·
1 Parent(s): 73c6f3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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, constrained_layout=True)
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
- # # 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)
 
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)