Spaces:

EgorDan
/

book-search

Runtime error

EgorDan commited on May 5, 2023

Commit

5963bd7

•

1 Parent(s): 234ebfe

deleted assignmetn model to weights

Files changed (1) hide show

pages/annotation_generator.py CHANGED Viewed

@@ -7,7 +7,7 @@ device = 'cpu'
 tokenizer = load('./pages/tokenizer.joblib')
 model = load('./pages/model.joblib')
-weights = model.load_state_dict(torch.load('./pages/model_weights.pt', map_location=device))
 temperature = st.slider('Градус дичи:', min_value = 1., max_value = 20., value = 3.)
 num_beams = st.slider('Число веток для поиска:', min_value = 1, max_value = 15, value = 7)

 tokenizer = load('./pages/tokenizer.joblib')
 model = load('./pages/model.joblib')
+model.load_state_dict(torch.load('./pages/model_weights.pt', map_location=device))
 temperature = st.slider('Градус дичи:', min_value = 1., max_value = 20., value = 3.)
 num_beams = st.slider('Число веток для поиска:', min_value = 1, max_value = 15, value = 7)