Update app.py

app.py

@@ -1,6 +1,6 @@
-!pip install -q -U trl transformers accelerate git+https://github.com/huggingface/peft.git
+pip install -q -U trl transformers accelerate git+https://github.com/huggingface/peft.git
 
-!pip install -q datasets bitsandbytes einops wandb
+pip install -q datasets bitsandbytes einops wandb
 
 
 from datasets import load_dataset
@@ -163,7 +163,7 @@ print(completion)
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-checkpoint_name="ArmelR/starcoder-gradio-v0"
+checkpoint_name= model
 model = AutoModelForCausalLM.from_pretrained(checkpoint_name)
 tokenizer = AutoTokenizer.from_pretrained(checkpoint_name)