awacke1 commited on
Commit
e168c81
·
verified ·
1 Parent(s): 2a63666

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -62,7 +62,7 @@ def save_image(img, prompt):
62
  'Hearts': [0],
63
  'Created': [datetime.now()]
64
  })
65
- image_metadata = pd.concat([image_metadata, new_row], ignore_index=True)
66
  likes_cache[filename] = {'likes': 0, 'dislikes': 0, 'hearts': 0}
67
  save_likes_cache(likes_cache)
68
  logger.info(f"Saved new image: {filename}")
@@ -163,6 +163,7 @@ def generate(
163
  guidance_scale: float = 3,
164
  randomize_seed: bool = False,
165
  progress=gr.Progress(track_tqdm=True),
 
166
  ):
167
  seed = int(randomize_seed_fn(seed, randomize_seed))
168
 
@@ -352,8 +353,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
352
  width,
353
  height,
354
  guidance_scale,
355
- randomize_seed,
356
- ],
357
  outputs=[result, seed, gr.HTML(visible=False), image_gallery, metadata_df],
358
  api_name="run",
359
  )
 
62
  'Hearts': [0],
63
  'Created': [datetime.now()]
64
  })
65
+ image_metadata = pd.concat([image_metadata, new_row], ignore_index=True, sort=False)
66
  likes_cache[filename] = {'likes': 0, 'dislikes': 0, 'hearts': 0}
67
  save_likes_cache(likes_cache)
68
  logger.info(f"Saved new image: {filename}")
 
163
  guidance_scale: float = 3,
164
  randomize_seed: bool = False,
165
  progress=gr.Progress(track_tqdm=True),
166
+ **kwargs # Add this line to accept any additional arguments
167
  ):
168
  seed = int(randomize_seed_fn(seed, randomize_seed))
169
 
 
353
  width,
354
  height,
355
  guidance_scale,
356
+ randomize_seed],
 
357
  outputs=[result, seed, gr.HTML(visible=False), image_gallery, metadata_df],
358
  api_name="run",
359
  )