se realiza cambio de nombre microsoft/Phi-3-mini-4k-instruct - CHATBOT

app.py

@@ -9,14 +9,14 @@ hf_token = os.getenv("HF_API_TOKEN")
 # Clase para manejar múltiples modelos
 class ModelHandler:
     def __init__(self, model_names, token):
-        self.clients = {model_name: InferenceClient(model_name, token=token) for model_name in model_names}
-        self.current_model = model_names[0]
+        self.clients = {model_key: InferenceClient(model_name, token=token) for model_key, model_name in model_names.items()}
+        self.current_model = list(model_names.keys())[0]
 
-    def switch_model(self, model_name):
-        if model_name in self.clients:
-            self.current_model = model_name
+    def switch_model(self, model_key):
+        if model_key in self.clients:
+            self.current_model = model_key
         else:
-            raise ValueError(f"Modelo {model_name} no está disponible.")
+            raise ValueError(f"Modelo {model_key} no está disponible.")
 
     def generate_response(self, input_text):
         prompt = f"Debes de responder a cualquier pregunta:\nPregunta: {input_text}"
@@ -31,10 +31,10 @@ class ModelHandler:
         except Exception as e:
             return f"Error al realizar la inferencia: {e}"
 
-# Lista de modelos disponibles
-model_names = [
-    "microsoft/Phi-3-mini-4k-instruct"
-]
+# Lista de modelos disponibles (con nombres amigables para la interfaz)
+model_names = {
+    "CHATBOT": "microsoft/Phi-3-mini-4k-instruct"
+}
 
 # Inicializa el manejador de modelos
 model_handler = ModelHandler(model_names, hf_token)
@@ -68,8 +68,8 @@ with gr.Blocks(title="Multi-Model LLM Chatbot with Image Generation") as demo:
     )
     with gr.Row():
         model_dropdown = gr.Dropdown(
-            choices=model_names + ["Generación de Imágenes"],
-            value=model_names[0],
+            choices=list(model_names.keys()) + ["Generación de Imágenes"],
+            value="CHATBOT",
             label="Seleccionar Acción/Modelo",
             interactive=True
         )