fffiloni commited on
Commit
60bf902
β€’
1 Parent(s): 0a0db87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
18
 
19
  def get_images(prompt):
20
  print("Calling SD")
21
- gallery_dir = stable_diffusion.predict(prompt, fn_index=2)
 
22
  img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
23
  return img_results[0]
24
 
18
 
19
  def get_images(prompt):
20
  print("Calling SD")
21
+ gallery_dir = stable_diffusion.predict(prompt, fn_index=2)
22
+ print(f"Gallery Directory: {gallery_dir}")
23
  img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
24
  return img_results[0]
25