Update app.py

app.py

@@ -29,7 +29,8 @@ def infer(prompt, init_image):
         num_inference_steps=50,
         generator=generator)
     
-    torch.cuda.empty_cache()
+    with torch.no_grad():
+        torch.cuda.empty_cache()
     
     res = pipe(alpha=1)