Spaces:

vives
/

bert_cvent_top_k_sim

Runtime error

vives commited on May 31, 2022

Commit

f563c99

1 Parent(s): 05fc1b3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from transformers import AutoModelForMaskedLM , AutoModelForSequenceClassification
 from transformers import AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
 import streamlit as st

+from transformers import AutoModelForMaskedLM , AutoModelForSequenceClassification, AutoModel
 from transformers import AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
 import streamlit as st