Update app.py

app.py

@@ -13,7 +13,12 @@ from sentence_transformers import SentenceTransformer, util
 from sklearn.metrics.pairwise import cosine_similarity
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-model = SentenceTransformer('stsb-distilbert-base', device=device)
+model = gr.interface.huggingface.load('stsb-distilbert-base') 
+
+#SentenceTransformer('stsb-distilbert-base', device=device)
+
+
+
 batch_size = 1
 scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)