Update app.py

app.py

@@ -2,6 +2,7 @@ from transformers import AutoModelForMaskedLM
 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)