Update app.py

app.py

@@ -16,7 +16,7 @@ import zipfile
 with zipfile.ZipFile("Vegetable Images.zip", 'r') as zip_ref:
     zip_ref.extractall('.')
 
-
+@st.cache_resource
 class ImageVectorizer:
     '''
     Get vector representation of an image using VGG19 model fine tuned on vegetable images for classification
@@ -41,7 +41,7 @@ class ImageVectorizer:
         test_image = np.array([test_image])
         return model(test_image).numpy()[0]
 
-
+@st.cache_resource
 def get_milvus_collection():
     uri = os.environ.get("URI")
     token = os.environ.get("TOKEN")