Update app.py

app.py

@@ -20,8 +20,8 @@ from sentence_transformers import SentenceTransformer, util
 
 #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)
 
@@ -53,14 +53,15 @@ def get_sim(x):
 	
 # Load pre-trained model 
 
-model = GPT2LMHeadModel.from_pretrained('distilgpt2', output_hidden_states = True, output_attentions = True)
-
+#model = GPT2LMHeadModel.from_pretrained('distilgpt2', output_hidden_states = True, output_attentions = True)
+model = GPT2LMHeadModel.from_pretrained('gpt2', output_hidden_states = True, output_attentions = True)
 #model  =  gr.Interface.load('huggingface/distilgpt2', output_hidden_states = True, output_attentions = True)
 
 #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')