Update app.py

app.py

@@ -21,14 +21,13 @@ def model_load():
 
 def sdxl_process(seed, prompt, additional_prompt, negative_prompt, num_steps, guidance_scale):
     pipe = model_load()
-    # generator = torch.Generator("cuda")
+    generator = torch.Generator("cuda")
     # generator.manual_seed(seed)
     
     special_prompt = 'jjj, scratch art style'
     prompt = f'{special_prompt}, {prompt}, with a black background'
     output = pipe(prompt, additional_prompt, negative_prompt=negative_prompt, num_inference_steps=num_steps, guidance_scale=guidance_scale,
-                # generator=generator).images[0]
-                ).images[0]
+                  generator=generator).images[0]
     
     return output