Spaces:

egumasa
/

engagement-analyzer-demo

Running

egumasa commited on Mar 11

Commit

73ef8c6

•

1 Parent(s): a93f8e6

bug fix

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -33,7 +33,8 @@ st.set_page_config(page_title="ENGAGEMENT analyzer (beta ver 0.3)",
 @st.cache(allow_output_mutation=True)
 def load_model():
     # nlp = spacy.load("en_engagement_RoBERTa_context_flz")
-    nlp = spacy.load("en_engagement_LSTM")
     return (nlp)

 @st.cache(allow_output_mutation=True)
 def load_model():
     # nlp = spacy.load("en_engagement_RoBERTa_context_flz")
+    # nlp = spacy.load("en_engagement_LSTM")
+    nlp = spacy.load("en_engagement_spl_RoBERTa_base_attention")
     return (nlp)