Spaces:
Sleeping
Sleeping
| import pickle | |
| import csv | |
| # Файл инициализации модели | |
| def init_model(): | |
| # Загрузить модели из файла | |
| pkl_filename = "pickle_model.pkl" | |
| with open(pkl_filename, 'rb') as file: | |
| model = pickle.load(file) | |
| # Считывание категорий | |
| file = open("cat.csv", "r") | |
| cat1 = list(csv.reader(file, delimiter=",")) | |
| categorias = cat1[0] | |
| file.close() | |
| model.eval() | |
| return model, categorias |