Update app.py

app.py

@@ -25,10 +25,10 @@ example = [
     [EXAMPLE_5, 32, 0.6, 42],
     ]
 
-device="cuda:0"
+
 tokenizer = AutoTokenizer.from_pretrained("loubnabnl/santacoder-code-to-text")
-model = AutoModelForCausalLM.from_pretrained("loubnabnl/santacoder-code-to-text", trust_remote_code=True).to(device)
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
+model = AutoModelForCausalLM.from_pretrained("loubnabnl/santacoder-code-to-text", trust_remote_code=True)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def make_doctring(gen_prompt):
     return gen_prompt + f"\n\n\"\"\"\nExplanation:"