Update gradio_app.py

gradio_app.py

@@ -96,6 +96,8 @@ def infer(frame1_path, frame2_path):
     frame2 = load_image(frame2_path)
     frame2 = frame2.resize((512, 288))
 
+    torch.cuda.empty_cache()
+
     frames = pipe(image1=frame1, image2=frame2, 
         num_inference_steps=num_inference_steps, # 50
         generator=generator,