added device support

app.py

@@ -5,8 +5,8 @@ from transformers import pipeline
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
 model_name = "eljanmahammadli/AzLlama-152M-Alpaca"
-model = pipeline("text-generation", model=model_name, torch_dtype=torch.float16).to(
-    device=device
+model = pipeline(
+    "text-generation", model=model_name, torch_dtype=torch.float16, device=device
 )
 logo_path = "AzLlama-logo.webp"