Update app.py

app.py

@@ -38,16 +38,50 @@ def inference(img):
     output = mainTest(inputpath, outputpath)
     print(time.strftime("Finish!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output
+from PIL import Image
+
+def load_image_from_file(file_path, new_height=None):
+    """
+    Load an image from a file and optionally resize it while maintaining the aspect ratio.
+
+    Args:
+        file_path (str): The path to the image file.
+        new_height (int, optional): The new height for the image. If None, the image is not resized.
+
+    Returns:
+        Image: The loaded (and optionally resized) image.
+    """
+    try:
+        img = Image.open(file_path)
+        
+        if new_height is not None:
+            # Calculate new width to maintain aspect ratio
+            aspect_ratio = img.width / img.height
+            new_width = int(new_height * aspect_ratio)
+            
+            # Resize the image
+            img = img.resize((new_width, new_height), Image.LANCZOS)
+        
+        return img
+    except FileNotFoundError:
+        print(f"File not found: {file_path}")
+        return None
+    except Image.UnidentifiedImageError:
+        print(f"Cannot identify image file: {file_path}")
+        return None
+    except Exception as e:
+        print(f"Error loading image from file: {e}")
+        return None
 
 title = "Undress AI"
 description = "⛔ Input photos of people, similar to the test picture at the bottom, and undress pictures will be produced. You may have to wait 30 seconds for a picture. 🔞 Do not upload personal photos 🔞 There is a queue system. According to the logic of first come, first served, only one picture will be made at a time. Must be able to at least see the outline of a human body ⛔"
 
 examples = [
-    ['example1.png'],
-    ['example2.png'],
-    ['example3.png'],
-    ['example5.webp'],
-    ['example6.webp'],
+    [load_image_from_file('example1.png')],
+    [load_image_from_file('example2.png')],
+    [load_image_from_file('example3.png')],
+    [load_image_from_file('example5.webp')],
+    [load_image_from_file('example6.webp')],
 ]
 
 css = """