Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed, pipeline
 
 #https://huggingface.co/spaces/lvwerra/codeparrot-generation
 
-title = "SantaCoder+Alpaca Generator 🎅🏾+🦙"
-description = "This is a subspace to make code generation with [SantaCoder](https://huggingface.co/bigcode/santacoder) and [Code Alpaca](https://github.com/sahil280114/codealpaca). Feel free to check this larger [space](https://huggingface.co/spaces/loubnabnl/Code-generation-models-v1) for more information about code generation with 🤗."
+title = "SantaCoder+Stack Exchange Generator 🎅🏾+📚"
+description = "This is a subspace to make code generation with [SantaCoder](https://huggingface.co/bigcode/santacoder) fine-tuned on [Stack Exchange](https://huggingface.co/datasets/ArmelR/stack-exchange-instruction). Feel free to check this larger [space](https://huggingface.co/spaces/loubnabnl/Code-generation-models-v1) for more information about code generation with 🤗."
 
 example = [
     ["def print_hello_world():", 8, 0.6, 42],
@@ -12,8 +12,8 @@ example = [
     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
 
-tokenizer = AutoTokenizer.from_pretrained("ArmelR/AlpacaCode512")
-model = AutoModelForCausalLM.from_pretrained("ArmelR/AlpacaCode512", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("ArmelR/Instruction10K2048")
+model = AutoModelForCausalLM.from_pretrained("ArmelR/Instruction10K2048", trust_remote_code=True)
 
 
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):