Add model and src folders

README.md

@@ -5,7 +5,7 @@ colorFrom: purple
 colorTo: green
 sdk: streamlit
 sdk_version: 1.29.0
-app_file: main.py
+app_file: src/main.py
 pinned: false
 ---
 

main.py → src/main.py

@@ -14,7 +14,7 @@ def load_recognition_model():
     """
     Loads the translation model pipeline.
     """
-    filename = "pickle_model.pkl"
+    filename = "model/pickle_model.pkl"
     with open(filename, 'rb') as file:
         model = pickle.load(file)
     return model

model_execute.py → src/model_execute.py

@@ -22,7 +22,7 @@ def output_to_names(output):
     """
     Converts model outputs to category names names.
     """
-    with open('cat.csv') as file:
+    with open('model/cat.csv') as file:
         reader = csv.reader(file)
         cat_list = list(reader)[0]
 
@@ -58,7 +58,7 @@ def check_photo(name, photo):
 
 
 if __name__ == "__main__":
-    pkl_filename = "pickle_model.pkl"
+    pkl_filename = "model/pickle_model.pkl"
     with open(pkl_filename, 'rb') as file:
         model = pickle.load(file)
 

model_retrain.py → src/model_retrain.py

@@ -173,7 +173,7 @@ model, optimizer, train_loader, valid_loader, 30)
 print('acc_train:', acc_train, '\nacc_valid:', acc_valid)
 
 # Сохранение модели в текущую рабочую директорию
-pkl_filename = "pickle_model.pkl"
+pkl_filename = "model/pickle_model.pkl"
 with open(pkl_filename, 'wb') as file:
     pickle.dump(model, file)