Spaces:

surkovvv
/

ysda-transformers-practice

Sleeping

surkovvv commited on Apr 17, 2023

Commit

6de5c29

•

1 Parent(s): be24c5d

added paths to model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ if st.button('Generate tags!'):
         st.write('# Please, enter title!')
     else:
         text = title + '[SEP]' + abstract
-        model, tokenizer = load_setup(path_to_model=, path_to_vocab=)
         prediction = predict(model, tokenizer, text)
         st.markdown(f'## Predicted tag(s): {prediction}')
 else:

         st.write('# Please, enter title!')
     else:
         text = title + '[SEP]' + abstract
+        model, tokenizer = load_setup(path_to_model='distilbert1.bin', path_to_vocab='vocab_distilbert1.bin')
         prediction = predict(model, tokenizer, text)
         st.markdown(f'## Predicted tag(s): {prediction}')
 else: