Update app.py

app.py

@@ -5,8 +5,12 @@ from typing import List, Tuple
 
 model_name = "Hawoly18/Adia_Llama3.1"
 
+# Vérifier si un GPU est disponible
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
 def respond(
     message: str,