Update app.py

app.py

@@ -38,7 +38,7 @@ trainer.save_model('./models', 'pytorch_model')
 src = './config.json'
 des = './models/config.json'
 os.rename(src, des)
-tokenizer = accelerator.prepare(AutoTokenizer.from_pretrained("./models"))
+tokenizer = accelerator.prepare(AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125m"))
 model = accelerator.prepare(AutoModelForCausalLM.from_pretrained("./models"))
 def plex(input_text):    
         mnputs = tokenizer(input_text, return_tensors='pt')