Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,6 @@ def make_prediction(text):
|
|
31 |
|
32 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
33 |
|
34 |
-
@st.cache
|
35 |
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
|
36 |
model_name = "trained_model2"
|
37 |
model_path = model_name + '.zip'
|
|
|
31 |
|
32 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
33 |
|
|
|
34 |
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
|
35 |
model_name = "trained_model2"
|
36 |
model_path = model_name + '.zip'
|