File size: 479 Bytes
8f186c9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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