Update app.py

app.py

@@ -1,18 +1,16 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed, pipeline
 
-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 removes files larger than 2MB in the current folder\nfind ."
-EXAMPLE_1 = "#!/bin/bash\n\n# This script send an email\nto=”admin@example.com”\nsubject=”Greeting”\nmsg=”Welcome to our site”\n"
-EXAMPLE_3 = "#!/bin/bash\n# This script convert avi files to mp4\nfor filename in $(ls *.avi); do\n"
-EXAMPLE_4 = "#!/bin/bash\nsource=$1\ndest=$2\n# copy source on dest\n"
-EXAMPLE_5 = """#!/bin/bash\n\n# This script check if the arg passed as first arg is a founder of huggingface\nfounders_array=("julien" "thom" "clem")"""
-
-
-examples = [[EXAMPLE_0, 14, 0.6, 42], [EXAMPLE_1, 28, 0.6, 42], [EXAMPLE_3, 46, 0.6, 42], [EXAMPLE_4, 35, 0.6, 43], [EXAMPLE_5, 70, 0.6, 43]]
-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)
+title = "SantaCoder 🎅 Dockerfiles 🐋 Completion"
+description = "This is a subspace to make code generation with [SantaCoder fine-tuned on The Stack Dockerfiles](https://huggingface.co/mrm8488/santacoder-finetuned-the-stack-dockerfiles)"
+EXAMPLE_0 = "# Dockerfile for Express API"
+
+
+CKPT = "mrm8488/santacoder-finetuned-the-stack-dockerfiles"
+
+examples = [[EXAMPLE_0, 55, 0.6, 42]]
+tokenizer = AutoTokenizer.from_pretrained(CKPT)
+model = AutoModelForCausalLM.from_pretrained(CKPT, trust_remote_code=True)
 
 
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):