Update app.py

app.py

@@ -10,7 +10,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
 )
 
 def generate_image(prompt):
-    with torch.grad():
+    with torch.no_grad():
         image = pipe(
             prompt=prompt,
             height=(height := 512),