chengzeyi commited on
Commit
b3b6622
·
1 Parent(s): 0e732d9
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -499,6 +499,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
499
  def get_recent_gallery_items():
500
  gallery_items = []
501
  for r in reversed(recent_generations):
 
 
502
  gallery_items.append((r["flux-dev"], f"FLUX-dev: {r['prompt']}"))
503
  gallery_items.append(
504
  (r["hidream-dev"], f"HiDream-dev: {r['prompt']}"))
 
499
  def get_recent_gallery_items():
500
  gallery_items = []
501
  for r in reversed(recent_generations):
502
+ if any(x is None for x in r.values()):
503
+ continue
504
  gallery_items.append((r["flux-dev"], f"FLUX-dev: {r['prompt']}"))
505
  gallery_items.append(
506
  (r["hidream-dev"], f"HiDream-dev: {r['prompt']}"))