added device support

app.py

@@ -2,8 +2,12 @@ import gradio as gr
 import torch
 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)
+model = pipeline("text-generation", model=model_name, torch_dtype=torch.float16).to(
+    device=device
+)
 logo_path = "AzLlama-logo.webp"