Fix memory leak

app.py

@@ -1,5 +1,6 @@
 from html import escape
 import re
+import torch
 import streamlit as st
 import pandas as pd, numpy as np
 from transformers import CLIPProcessor, CLIPModel, FlavaModel, FlavaProcessor
@@ -43,7 +44,8 @@ source = {0: "\nSource: Unsplash", 1: "\nSource: The Movie Database (TMDB)"}
 
 def compute_text_embeddings(list_of_strings, name):
     inputs = processors[name](text=list_of_strings, return_tensors="pt", padding=True)
-    result = models[name].get_text_features(**inputs)
+    with torch.no_grad():
+        result = models[name].get_text_features(**inputs)
     if "flava" in name:
         result = result[:, 0, :]
     result = result.detach().numpy()