Update app.py

app.py

@@ -6,11 +6,9 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 title = "Santacoder bash/shell Completion 🎅"
 description = "This is a subspace to make code generation with [SantaCoder fine-tuned on The Stack bash/shell](https://huggingface.co/mrm8488/santacoder-finetuned-the-stack-bash-4)"
+EXAMPLE_0 = "#!/bin/bash\n# This script read the files in the current dir and remove it if greater than 2MB"
 example = [
-    ["def print_hello_world():", 8, 0.6, 42],
-    ["def get_file_size(filepath):", 40, 0.6, 42],
-    ["def count_lines(filename):", 40, 0.6, 42],
-    ["def count_words(filename):", 40, 0.6, 42]]
+    [EXAMPLE_0, 128, 0.6, 42]]
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/santacoder-finetuned-the-stack-bash-4")
 model = AutoModelForCausalLM.from_pretrained("mrm8488/santacoder-finetuned-the-stack-bash-4", trust_remote_code=True).to(device)