Spaces:

vives
/

bert_sentence_sim

Runtime error

vives commited on May 13, 2022

Commit

77b655a

1 Parent(s): f8b1359

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ from transformers import AutoModelForMaskedLM
 from transformers import AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
 import streamlit as st
 model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
 model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)

 from transformers import AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
 import streamlit as st
+import torch
 model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
 model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)