fffiloni commited on
Commit
072c110
·
1 Parent(s): 634ecf0

Update app.py

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