Update app.py

app.py

@@ -21,7 +21,7 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
 
 model_id1 = os.getenv("API_KEY")
-pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch_dtype, use_safetensors=True)
+pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True)
 pipe = pipe.to('cpu')
 
 # Updated function for text generation using the new API structure