Update app.py

app.py

@@ -11,12 +11,13 @@ from pathos.multiprocessing import ProcessingPool as ProcessPoolExecutor
 import requests
 from lxml.html import fromstring
 
-pool = ProcessPoolExecutor(1000)
+pool = ProcessPoolExecutor(8)
 pool.__enter__()
 
+#model_id = "runwayml/stable-diffusion-v1-5"
 model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
-device = "cuda" if torch.cuda.is_available() else "cpu"
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
@@ -72,7 +73,7 @@ def infer(prompt):
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 12cm;
+    max-width: 16cm;
 }
 #image-container {
     aspect-ratio: 1 / 1;