seawolf2357 commited on
Commit
f7ae27d
·
verified ·
1 Parent(s): 3a529fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -65,15 +65,15 @@ def load_examples():
65
  "a bustling market in a small town"
66
  ]
67
 
68
- for i, filename in enumerate(["00.png", "02.png", "03.png"]):
69
  img_path = os.path.join(test_img_dir, filename)
70
  if os.path.exists(img_path):
71
  # Use dimensions from original code for each specific example
72
- if filename == "00.png":
73
  height, width = 680, 1024
74
- elif filename == "02.png":
75
  height, width = 560, 1024
76
- elif filename == "03.png":
77
  height, width = 568, 1024
78
  else:
79
  height, width = 768, 768
 
65
  "a bustling market in a small town"
66
  ]
67
 
68
+ for i, filename in enumerate(["00.jpg", "02.jpg", "03.jpg]):
69
  img_path = os.path.join(test_img_dir, filename)
70
  if os.path.exists(img_path):
71
  # Use dimensions from original code for each specific example
72
+ if filename == "00.jpg":
73
  height, width = 680, 1024
74
+ elif filename == "02.jpg":
75
  height, width = 560, 1024
76
+ elif filename == "03.jpg":
77
  height, width = 568, 1024
78
  else:
79
  height, width = 768, 768