Update app.py

app.py

@@ -19,7 +19,7 @@ example = [
 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=device)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
 
 def make_doctring(gen_prompt):
     return gen_prompt + f"\n\n\"\"\"\nExplanation:"