debug urls

app.py

@@ -103,6 +103,27 @@ def main(
     n_samples=4,
     ):
     debug_print("main")
+    images = []
+    for url in test_images_urls:
+        import requests
+        from io import BytesIO
+        from PIL import Image
+        try:
+            response = requests.get(url)
+            if not response.ok:
+                continue
+            bytes = BytesIO(response.content)
+            image = Image.open(bytes)
+            if image.mode != 'RGB':
+                image = image.convert('RGB')
+            # width = 336
+            # aspect_ratio = float(image.height) / float(image.width)
+            # height = int(width * aspect_ratio)
+            # image = image.resize((width, height), Image.Resampling.LANCZOS)
+            images.append((image, "title"))
+        except Exception as e:
+            print(e)
+    return images    
 
     embeddings = base64_to_embedding(embeddings)
     # convert to python array
@@ -282,6 +303,12 @@ image_folder ="images"
 #         "Prompt 3": "a person sitting on a bench using a cell phone",
 
 # }
+test_images_urls = [
+    "https://www.mdig.com.br/imagens/bichos/caes_miniatura_chicara_07.jpg",
+    "https://i.pinimg.com/236x/04/ac/0e/04ac0e05964b75c9db59de94c571339e.jpg",
+    "https://i.pinimg.com/236x/a1/55/9e/a1559e56ae5fb6e4d19c43c6396b5940--teacup-yorkie-yorkie-puppy.jpg",
+    "https://i.pinimg.com/236x/11/c4/51/11c4518febb0869bd2b2d391d0753f44.jpg",
+]
 
 tabbed_examples = {
     "Pups": {