Update app.py

app.py

@@ -10,7 +10,7 @@ import requests
 #url = "https://github.com/simonepri/lm-scorer/tree/master/lm_scorer/models"
 #resp = requests.get(url)
 
-from sentence_transformers import SentenceTransformer, util
+#from sentence_transformers import SentenceTransformer, util
 #from sentence_transformers import SentenceTransformer, util
 #from sklearn.metrics.pairwise import cosine_similarity
 #from lm_scorer.models.auto import AutoLMScorer as LMScorer
@@ -18,7 +18,7 @@ from sentence_transformers import SentenceTransformer, util
 #from sklearn.metrics.pairwise import cosine_similarity
 
 
-#model_sts = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
+model_sts = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
 
 model_sts = SentenceTransformer('stsb-distilbert-base')
 
@@ -55,10 +55,10 @@ def get_sim(x):
 
 model = GPT2LMHeadModel.from_pretrained('distilgpt2', output_hidden_states = True, output_attentions = True)
 
-#model  =  gr.Interface.load('huggingface/distilgpt2', 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 =  gr.Interface.load('huggingface/distilgpt2')
 
 tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')