update app.py

app.py

@@ -5,8 +5,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 # checkpoint = "Salesforce/codegen-2B-nl"
 checkpoint = "Salesforce/codegen2-1B"
 
-# tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(checkpoint, cache_dir="models/")
+tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
+# model = AutoModelForCausalLM.from_pretrained(checkpoint, cache_dir="models/")
 model = AutoModelForCausalLM.from_pretrained(checkpoint)
 
 def code_gen(text):