Update app.py

app.py

@@ -20,8 +20,8 @@ from sentence_transformers import SentenceTransformer, util
 #device = "cuda:0" if torch.cuda.is_available() else "cpu"
 #model_sts = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
 
-#model_sts = SentenceTransformer('stsb-distilbert-base')
-model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
+model_sts = SentenceTransformer('stsb-distilbert-base')
+#model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
 
 #batch_size = 1
 #scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)