Update app.py

app.py

@@ -26,6 +26,9 @@ def set_predictor(image):
     predictor = SamPredictor(sam)
     predictor.set_image(image)
 
+    if device == 'cuda':
+        torch.cuda.empty_cache()
+
     return [image, predictor, 'Done']