Update app.py

app.py

@@ -14,7 +14,6 @@ pipe = DiffusionPipeline.from_pretrained(
     safety_checker=None,
     use_auth_token=MY_SECRET_TOKEN,
     revision="fp16",
-    torch_dtype=torch.float16,
     custom_pipeline="imagic_stable_diffusion",
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)
@@ -32,6 +31,8 @@ def infer(prompt, init_image):
         num_inference_steps=50,
         generator=generator)
     
+    torch.cuda.empty_cache()
+    
     res = pipe(alpha=1)
 
     return res.images[0]