Update app.py

app.py

@@ -39,7 +39,7 @@ def sample_model():
    del unet
    global network
    unet, _, _, _, _ = load_models(device)
-
+@torch.no_grad()
 def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     global device
     global generator