fix: close read file

utils.py

@@ -9,13 +9,13 @@ from koclip import FlaxHybridCLIP
 @st.cache(allow_output_mutation=True)
 def load_index(img_file):
     filenames, embeddings = [], []
-    lines = open(img_file, "r")
-    for line in lines:
-        cols = line.strip().split("\t")
-        filename = cols[0]
-        embedding = [float(x) for x in cols[1].split(",")]
-        filenames.append(filename)
-        embeddings.append(embedding)
+    with open(img_file, "r") as f:
+        for line in f:
+            cols = line.strip().split("\t")
+            filename = cols[0]
+            embedding = [float(x) for x in cols[1].split(",")]
+            filenames.append(filename)
+            embeddings.append(embedding)
     embeddings = np.array(embeddings)
     index = nmslib.init(method="hnsw", space="cosinesimil")
     index.addDataPointBatch(embeddings)
@@ -34,10 +34,3 @@ def load_model(model_name="koclip/koclip-base"):
             "google/vit-large-patch16-224"
         )
     return model, processor
-
-
-@st.cache(allow_output_mutation=True)
-def load_model_v2(model_name="koclip/koclip"):
-    model = FlaxHybridCLIP.from_pretrained(model_name)
-    processor = CLIPProcessor.from_pretrained(model_name)
-    return model, processor