Spaces:

1NEYRON1
/

Topic_classification_for_scientific_articles

Running

1NEYRON1 commited on Apr 7

Commit

e071fb5

1 Parent(s): 1b304ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,6 @@ id_to_cat = {
     137: 'Algebraic Geometry'
 }
-@st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
     model = AutoModelForSequenceClassification.from_pretrained(

     137: 'Algebraic Geometry'
 }
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
     model = AutoModelForSequenceClassification.from_pretrained(