Spaces:

dfedukov
/

papers_classifier

Sleeping

dfedukov commited on 23 days ago

Commit

c0a3229

verified ·

1 Parent(s): ccf784a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,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}

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