Update app.py

app.py

@@ -35,26 +35,17 @@ def infer(prompt):
     #generator = torch.Generator(device=device).manual_seed(seed)
     #print("Is GPU busy? ", is_gpu_busy)
     images = []
-    #if(not is_gpu_busy):
-    #    is_gpu_busy = True
-    #    images_list = pipe(
-    #        [prompt] * samples,
-    #        num_inference_steps=steps,
-    #        guidance_scale=scale,
-            #generator=generator,
-    #    )
-    #    is_gpu_busy = False
-    #    safe_image = Image.open(r"unsafe.png")
-    #    for i, image in enumerate(images_list["sample"]):
-     #       if(images_list["nsfw_content_detected"][i]):
-     #           images.append(safe_image)
-     #       else:
-     #           images.append(image)
-    #else:
-    with autocast("cuda"):
-        generated = pipe(prompt, guidance_scale=7.5)["sample"]
-    
-    return generated
+    if(not is_gpu_busy):
+        is_gpu_busy = True
+        with autocast("cuda"):
+            images = pipe(
+                [prompt] * samples,
+                num_inference_steps=steps,
+                guidance_scale=scale,
+                #generator=generator,
+            ).images
+
+    return images
     
     
 css = """