update app

app.py

@@ -7,8 +7,7 @@ title = "CodeParrot Generator 🦜"
 description = "This is a subspace to make code generation with CodeParrot, it is used in a larger space for model comparison."
 example = [
     ["def print_hello_world():", "Sample", 8, 42],
-    ["\"""import GPT2 from transformers\""" ", "Sample", 16, 42],
-    ["def get_file_size(filepath):", "Sample", 8, 42]]
+    ["def get_file_size(filepath):", "Sample", 22, 42]]
 tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
 model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage=True)