Update app.py

app.py

@@ -23,10 +23,7 @@ Chat with Buzz-small!
 only 3b,  this demo runs on the fp16 weights of the model in safetensors format (converting to cpp soon)
 """
 
-if torch.cuda.is_available():
-    torch.device('cuda')
-else:
-    torch.device('cpu')
+device = 0 if torch.cuda.is_available() else -1
 
 model_id = "H-D-T/Buzz-3b-small-v0.6.3"
 chatbot = pipeline(model=model_id, device=device, task="conversational")