awacke1 commited on
Commit
67bddd7
1 Parent(s): c89f94f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def predict(prompt, guidance, steps, seed=1231231):
90
  if len(results.images) > 0:
91
  image_path = os.path.join("path_to_save", filename) # Specify your directory
92
  results.images[0].save(image_path)
93
- print(f"Image saved as {image_path}")
94
 
95
  return results.images[0] if len(results.images) > 0 else None
96
 
 
90
  if len(results.images) > 0:
91
  image_path = os.path.join("path_to_save", filename) # Specify your directory
92
  results.images[0].save(image_path)
93
+ gr.Markdown(f"#Image saved as {image_path}")
94
 
95
  return results.images[0] if len(results.images) > 0 else None
96