Update app.py

app.py

@@ -20,7 +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('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)
@@ -50,7 +51,8 @@ model = GPT2LMHeadModel.from_pretrained('distilgpt2', output_hidden_states = Tru
 #model.eval()
 #tokenizer =  gr.Interface.load('huggingface/distilgpt2')
 
-tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
+#tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')