Update app.py

app.py

@@ -10,8 +10,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
 )
 
 def generate_image(prompt):
-    with torch.device(device):
-        print(f"Current device: {torch.device(device)}")
+    with torch.grad():
         image = pipe(
             prompt=prompt,
             height=(height := 512),