Update app.py

app.py

@@ -10,7 +10,7 @@ EXAMPLE_4 = "#!/bin/bash\nsource=$1\ndest=$2\n"
 
 examples = [[EXAMPLE_0, 14, 0.6, 42], [EXAMPLE_1, 28, 0.6, 42], [EXAMPLE_3, 46, 0.6, 42], [EXAMPLE_4, 46, 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).to(device)
+model = AutoModelForCausalLM.from_pretrained("mrm8488/santacoder-finetuned-the-stack-bash-4", trust_remote_code=True).to("cuda")
 
 
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):