Update app.py

app.py

@@ -12,8 +12,9 @@ example = [
     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
 
-tokenizer = AutoTokenizer.from_pretrained("ArmelR/Instruction10K2048")
-model = AutoModelForCausalLM.from_pretrained("ArmelR/Instruction10K2048", trust_remote_code=True)
+checkpoint = "ArmelR/Instruction10K2048"
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True)
 
 
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):