Added: empty_cache (#2)

- Added: empty_cache (c86c3e08ffdbf43266c7f4d2bb294b97cc09b1e0)


Co-authored-by: Max Skobeev <[email protected]>

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