AMfeta99 commited on
Commit
cbc8cbc
·
verified ·
1 Parent(s): a778509

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def add_label_to_image(image, label):
49
  def plot_and_save_agent_image(agent_image, label, save_path=None):
50
  pil_image = agent_image.to_raw()
51
  labeled_image = add_label_to_image(pil_image, label)
52
- labeled_image.show()
53
  if save_path:
54
  labeled_image.save(save_path)
55
  print(f"Image saved to {save_path}")
 
49
  def plot_and_save_agent_image(agent_image, label, save_path=None):
50
  pil_image = agent_image.to_raw()
51
  labeled_image = add_label_to_image(pil_image, label)
52
+ #labeled_image.show()
53
  if save_path:
54
  labeled_image.save(save_path)
55
  print(f"Image saved to {save_path}")