Merge pull request #3 from kavlab/kuznetsov_av

.gitignore

@@ -146,3 +146,5 @@ dmypy.json
 # Cython debug symbols
 cython_debug/
 
+# Streamlit
+.devcontainer

README.md

@@ -19,7 +19,7 @@
 ## Как запустить
 Запуск осуществляется через модуль streamlit:
 ```
-streamlit run <путь>\run.py
+streamlit run run.py
 ```
 
 ## Как использовать

run.py

@@ -39,8 +39,8 @@ def main_app() -> None:
     """
 
     st_page_rename({"run.py": "Главная страница",
-                    "pages\page_one.py": "Генератор аудио",
-                    "pages\page_two.py": "Описание изображения"})
+                    "pages/page_one.py": "Генератор аудио",
+                    "pages/page_two.py": "Описание изображения"})
 
     st.markdown(read_readme(), unsafe_allow_html=True)
 

zvereva_ev/get_description_image.py

@@ -26,12 +26,18 @@ def load_image():
         return None
 
 
+@st.cache_resource
+def load_models():
+    processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
+    model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
+    return processor, model
+
+
 def get_description_image():
     """
     Функция get_description_image() открывает загруженное изображение пользователем и генерирует описание его
     """
-    processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
-    model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
+    processor, model = load_models()
 
     raw_image = Image.open("image.png")