Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ model.load_state_dict(torch.load('weight_model'))
|
|
19 |
|
20 |
text = get_text(title, abstract)
|
21 |
if text:
|
22 |
-
raw_predictions =
|
23 |
st.markdown(f"{raw_predictions}")
|
24 |
else:
|
25 |
st.markdown("Ваш запрос пуст. Введите хотя бы название")
|
|
|
19 |
|
20 |
text = get_text(title, abstract)
|
21 |
if text:
|
22 |
+
raw_predictions = get_labels(text, model, tokenizer)
|
23 |
st.markdown(f"{raw_predictions}")
|
24 |
else:
|
25 |
st.markdown("Ваш запрос пуст. Введите хотя бы название")
|