Update app.py

app.py

@@ -10,8 +10,8 @@ import time
 repo = "stabilityai/stable-diffusion-xl-base-1.0"
 trigger_word = "T shirt design, TshirtDesignAF, "
 
-def generate_image(prompt):
-    print("Generating image with prompt:", prompt)
+def generate_images(prompt):
+    print("Generating images with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
     #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
     headers = {
@@ -27,33 +27,39 @@ def generate_image(prompt):
         },
     }
 
-    error_count = 0
-    pbar = tqdm(total=None, desc="Loading model")
-    while True:
-        print("Sending request to API...")
-        response = requests.post(api_url, headers=headers, json=payload)
-        print("API response status code:", response.status_code)
-        if response.status_code == 200:
-            print("Image generation successful!")
-            return Image.open(BytesIO(response.content))  # Changed to match the first code
-        elif response.status_code == 503:
-            time.sleep(1)
-            pbar.update(1)
-        elif response.status_code == 500 and error_count < 5:
-            time.sleep(1)
-            error_count += 1
-        else:
-            print("API Error:", response.status_code)
-            raise Exception(f"API Error: {response.status_code}")
+    images = []
+    for i in range(10):
+        error_count = 0
+        pbar = tqdm(total=None, desc=f"Loading model {i+1}/10")
+        while True:
+            print(f"Sending request to API for image {i+1}...")
+            response = requests.post(api_url, headers=headers, json=payload)
+            print("API response status code:", response.status_code)
+            if response.status_code == 200:
+                print(f"Image {i+1} generation successful!")
+                img = Image.open(BytesIO(response.content))
+                images.append(img)
+                break
+            elif response.status_code == 503:
+                time.sleep(1)
+                pbar.update(1)
+            elif response.status_code == 500 and error_count < 5:
+                time.sleep(1)
+                error_count += 1
+            else:
+                print(f"API Error for image {i+1}: {response.status_code}")
+                raise Exception(f"API Error: {response.status_code}")
+
+    return images
 
 iface = gr.Interface(
-    fn=generate_image,
+    fn=generate_images,
     inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),
-    outputs="image",
-    title="Desgin by rahul7star",
+    outputs=gr.Gallery(label="Generated Images").style(grid=[5]),  # Display images in a grid of 5 columns
+    title="Design by rahul7star",
     description="Make designs for your clothes",
     examples=[["Cute Panda"], ["Skull"]]
 )
 
 print("Launching Gradio interface...")
-iface.launch()
+iface.launch()