Update app.py

app.py

@@ -7,10 +7,13 @@ import os
 import gradio as gr
 import requests
 import torch
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
+#from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from torch.nn.functional import softmax
 import numpy as np
 
+from transformers import AutoTokenizer, AutoModelForCausalLM
+#from torch.nn.functional import softmax
+
 # just for the sake of this demo, we use cloze prob to initialize the hypothesis 
 
 #url = "https://github.com/simonepri/lm-scorer/tree/master/lm_scorer/models"