change checkpoint to Salesforce/codegen-2B-nl

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen25-7b-instruct", trust_remote_code=True)
-# model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen25-7b-instruct", cache_dir="models/")
-model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen25-7b-instruct")
+# checkpoint = "Salesforce/codegen25-7b-instruct"
+checkpoint = "Salesforce/codegen-2B-nl"
+
+# 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):
     input_ids = tokenizer(text, return_tensors="pt").input_ids

requirements.txt

@@ -1,4 +1,4 @@
 transformers
-tiktoken
+# tiktoken
 torch
 torchvision