stefania11 commited on
Commit
500d7f4
·
1 Parent(s): 07e0495

update app file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def get_images(prompt):
30
  return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
31
 
32
  def get_code(prompt):
33
- examples_dir = code_generation(prompt, fn_index=2)
34
  return [os.path.join(examples_dir, output) for output in os.listdir(examples_dir)]
35
 
36
 
 
30
  return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
31
 
32
  def get_code(prompt):
33
+ examples_dir = code_generation(prompt, fn_index=0)
34
  return [os.path.join(examples_dir, output) for output in os.listdir(examples_dir)]
35
 
36