Upload folder using huggingface_hub

index.py

@@ -78,35 +78,17 @@ def image_classifier(prompt, starter_image, image_strength):
 
     print(output)
     
-    # Download the image from the URL
-    image_url = output[0]
-    print(image_url)
-    response = requests.get(image_url)
-    print(response)
-    img1 = Image.open(io.BytesIO(response.content))
-
-    # Download the image from the URL
-    image_url = output[1]
-    print(image_url)
-    response = requests.get(image_url)
-    print(response)
-    img2 = Image.open(io.BytesIO(response.content))
-
-    # Download the image from the URL
-    image_url = output[2]
-    print(image_url)
-    response = requests.get(image_url)
-    print(response)
-    img3 = Image.open(io.BytesIO(response.content))
-
-    return [img1, img2, img3]
-
-
-# app = Flask(__name__)
-# os.environ.get("REPLICATE_API_TOKEN")
+    images = []
+    for i in range(min(len(output), 3)):
+        image_url = output[i]
+        response = requests.get(image_url)
+        images.append(Image.open(io.BytesIO(response.content)))
+    
+    # Add empty images if fewer than 3 were returned
+    while len(images) < 3:
+        images.append(Image.new('RGB', (512, 512), 'gray'))
 
-# @app.route("/")
-# def index():
+    return images
 
 demo = gr.Interface(fn=image_classifier, inputs=["text", "image", gr.Slider(0, 1, step=0.025, value=0.2, label="Image Strength")], outputs=["image", "image", "image"])
 demo.launch(share=False)