Amitontheweb commited on
Commit
4dcdb6a
·
verified ·
1 Parent(s): 6c21e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -160,8 +160,8 @@ def select_model(model_selected):
160
  def load_model ():
161
 
162
  global model_name
163
- tokenizer_gpt2 = AutoTokenizer.from_pretrained(model_name)
164
- model_gpt2 = AutoModelForCausalLM.from_pretrained(model_name)
165
 
166
 
167
 
 
160
  def load_model ():
161
 
162
  global model_name
163
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
164
+ model = AutoModelForCausalLM.from_pretrained(model_name)
165
 
166
 
167