update app

app.py

@@ -6,8 +6,8 @@ from transformers import pipeline
 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():\n \"""Print 'Hello world' \""" "
-tokenizer = load_tokenizer("lvwerra/codeparrot")
-model = load_model("lvwerra/codeparrot")
+tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
+model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage=True)
     
 
 def code_generation(gen_prompt, strategy, max_tokens, seed=42):
@@ -24,7 +24,7 @@ def code_generation(gen_prompt, strategy, max_tokens, seed=42):
     return generated_text
 
 
-interface = gr.Interface(
+iface = gr.Interface(
     fn=code_generation, 
     inputs=[
         gr.Textbox(lines=10, label="Input code"),
@@ -51,4 +51,4 @@ interface = gr.Interface(
     description=description,
     title=title
 )
-interface.launch()
+iface.launch()