Update app.py

app.py

@@ -11,24 +11,25 @@ repo = "stabilityai/stable-diffusion-xl-base-1.0"
 trigger_word = "T shirt design, TshirtDesignAF, "
 
 def generate_images(prompt):
-    print("Generating images with prompt:", prompt)
+    print("Generating 10 unique 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 = {
         #"Authorization": f"Bearer {token}"
     }
-    full_prompt = f"{prompt} {trigger_word}"
-    payload = {
-        "inputs": full_prompt,
-        "parameters": {
-            "negative_prompt": "(worst quality, low quality, normal quality, lowres, low details, oversaturated, undersaturated, overexposed, underexposed, grayscale, bw, bad photo, bad photography, bad art:1.4), (watermark, signature, text font, username, error, logo, words, letters, digits, autograph, trademark, name:1.2), (blur, blurry, grainy), morbid, ugly, asymmetrical, mutated malformed, mutilated, poorly lit, bad shadow, draft, cropped, out of frame, cut off, censored, jpeg artifacts, out of focus, glitch, duplicate, (airbrushed, cartoon, anime, semi-realistic, cgi, render, blender, digital art, manga, amateur:1.3), (3D ,3D Game, 3D Game Scene, 3D Character:1.1), (bad hands, bad anatomy, bad body, bad face, bad teeth, bad arms, bad legs, deformities:1.3)",
-            "num_inference_steps": 30,
-            "scheduler": "DPMSolverMultistepScheduler"
-        },
-    }
-
+    
     images = []
     for i in range(10):
+        full_prompt = f"{prompt} {trigger_word} unique_design_{i}"
+        payload = {
+            "inputs": full_prompt,
+            "parameters": {
+                "negative_prompt": "(worst quality, low quality, normal quality, lowres, low details, oversaturated, undersaturated, overexposed, underexposed, grayscale, bw, bad photo, bad photography, bad art:1.4), (watermark, signature, text font, username, error, logo, words, letters, digits, autograph, trademark, name:1.2), (blur, blurry, grainy), morbid, ugly, asymmetrical, mutated malformed, mutilated, poorly lit, bad shadow, draft, cropped, out of frame, cut off, censored, jpeg artifacts, out of focus, glitch, duplicate, (airbrushed, cartoon, anime, semi-realistic, cgi, render, blender, digital art, manga, amateur:1.3), (3D ,3D Game, 3D Game Scene, 3D Character:1.1), (bad hands, bad anatomy, bad body, bad face, bad teeth, bad arms, bad legs, deformities:1.3)",
+                "num_inference_steps": 30,
+                "scheduler": "DPMSolverMultistepScheduler"
+            },
+        }
+
         error_count = 0
         pbar = tqdm(total=None, desc=f"Loading model {i+1}/10")
         while True: