Added: empty_cache

app.py

@@ -25,6 +25,9 @@ def generate_caption(image, caption_type):
 
     caption = "\n".join(caption)
 
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+
     return caption