Update app.py

app.py

@@ -4,12 +4,14 @@ import gradio
 # Especificar el dispositivo en el que se cargará el modelo (en este caso, "cuda" para GPU)
 device = "cuda"
 
+model_name = 'mistralai/Mistral-7B-Instruct-v0.1'
+
 # Cargar el modelo preentrenado y el tokenizador asociado
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 
 # Cargar el tokenizador asociado
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 def speak(prompt):
     # Tokenizar el prompt y convertirlo a tensores de PyTorch, luego enviarlos al dispositivo especificado