Update app.py

app.py

@@ -9,12 +9,10 @@ from diffusers import StableDiffusion3Pipeline
 device = 'cuda' #if torch.cuda.is_available() else 'cpu'
 torch.cuda.max_memory_allocated(device=device)
 torch.cuda.empty_cache()
-SD3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
+SD3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16).to(device)
 SD3.enable_xformers_memory_efficient_attention()
-SD3 = SD3.to(device)
-
                     
-def genie (Prompt, negative_prompt, height, width, scale, steps, d_steps, seed):
+def genie (Prompt, negative_prompt, height, width, scale, steps, seed):
     generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
     torch.cuda.empty_cache()
     image=SD3(