Spaces:
Runtime error
Runtime error
SarmadBashir
commited on
Commit
β’
8da0593
1
Parent(s):
ef8be1a
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ st.set_page_config(
|
|
8 |
#page_icon="π",
|
9 |
)
|
10 |
|
11 |
-
@st.cache
|
12 |
def get_pipleine():
|
13 |
tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
|
14 |
model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")
|
|
|
8 |
#page_icon="π",
|
9 |
)
|
10 |
|
11 |
+
@st.cache(allow_output_mutation=True)
|
12 |
def get_pipleine():
|
13 |
tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
|
14 |
model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")
|