Update app.py

app.py

@@ -69,22 +69,21 @@ def load_examples():
         img_path = os.path.join(test_img_dir, filename)
         if os.path.exists(img_path):
             # Use dimensions from original code for each specific example
-            if filename == "00.jpg":
+            if filename == "00.png":
                 height, width = 680, 1024
-            elif filename == "02.jpg":
+            elif filename == "02.png":
                 height, width = 560, 1024
-            elif filename == "03.jpg":
+            elif filename == "03.png":
                 height, width = 568, 1024
-
             else:
                 height, width = 768, 768
                 
             examples.append([
                 example_prompts[i % len(example_prompts)],  # User prompt (without system prompt)
                 Image.open(img_path),                       # Reference image
-                height,                                    # Height
-                width,                                     # Width
-                i + 1                                      # Seed
+                height,                                     # Height
+                width,                                      # Width
+                i + 1                                       # Seed
             ])
     
     return examples