Spaces:

dfedukov
/

papers_classifier

Sleeping

dfedukov commited on 24 days ago

Commit

4b87ee5

verified ·

1 Parent(s): b405130

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,6 @@
 import streamlit as st
 import torch
-from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
 target_to_ind = {'cs': 0, 'econ': 1, 'eess': 2, 'math': 3, 'phys': 4, 'q-bio': 5, 'q-fin': 6, 'stat': 7}

 import streamlit as st
 import torch
+from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification
 target_to_ind = {'cs': 0, 'econ': 1, 'eess': 2, 'math': 3, 'phys': 4, 'q-bio': 5, 'q-fin': 6, 'stat': 7}