Update app.py

app.py

@@ -55,13 +55,13 @@ model = GPT2LMHeadModel.from_pretrained('gpt2', output_hidden_states = True, out
 #tokenizer =  gr.Interface.load('huggingface/distilgpt2')
 
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 
 
 
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2LMHeadModel.from_pretrained('gpt2')
+tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
+model = GPT2LMHeadModel.from_pretrained('distilgpt2')
 
 def sentence_prob_mean(text):
     # Tokenize the input text and add special tokens