Update CaptionGenerator.py

CaptionGenerator.py

@@ -117,15 +117,15 @@ class CaptionGenerator:
             st.error(f"Error occurred with Gemini API: {e}")
 
 
-@st.cache_resource()
-def load_model():
-    """
-    Loads the BLIP-2 model for image captioning. This function is cached to avoid
-    re-loading the model on every call.
-    :param CaptionGenerator:  The class for generating captions for images.
-    :return: Instance of the CaptionGenerator class.
-    """
-    return CaptionGenerator()
+# @st.cache_resource()
+# def load_model():
+#     """
+#     Loads the BLIP-2 model for image captioning. This function is cached to avoid
+#     re-loading the model on every call.
+#     :param CaptionGenerator:  The class for generating captions for images.
+#     :return: Instance of the CaptionGenerator class.
+#     """
+#     return CaptionGenerator()
 
 
 # Example usage of the CaptionGenerator class