Spaces:

SarmadBashir
/

REFSQ2023_ReqORNot_demo_app

Runtime error

SarmadBashir commited on Jul 14, 2023

Commit

4bd96e0

•

1 Parent(s): b655d81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ st.set_page_config(
     #page_icon="🎈",
 )
-@st.cache(allow_output_mutation=True)
 def get_pipleine():
     tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
     model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")

     #page_icon="🎈",
 )
+st.cache_resource
 def get_pipleine():
     tokenizer = AutoTokenizer.from_pretrained("SarmadBashir/dronology_bert_uncased", model_max_length = 128)
     model = AutoModelForSequenceClassification.from_pretrained("SarmadBashir/dronology_bert_uncased")