Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Cargar el modelo y el tokenizador
-model_name = "meta-llama/Llama-2-7b-chat-hf"
+model_name = "facebook/opt-1.3b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", load_in_8bit=True)
 
@@ -26,11 +26,11 @@ def generate_response(prompt, max_length=200):
 def chatbot(message, history):
     history = history or []
     
-    # Construir el prompt en el formato que Llama-2 espera
+    # Construir el prompt
     prompt = "Eres un asistente AI amigable y útil. Responde de manera concisa y coherente.\n\n"
     for human, ai in history:
-        prompt += f"Human: {human}\nAssistant: {ai}\n"
-    prompt += f"Human: {message}\nAssistant:"
+        prompt += f"Human: {human}\nAI: {ai}\n"
+    prompt += f"Human: {message}\nAI:"
     
     response = generate_response(prompt)
     
@@ -41,8 +41,8 @@ iface = gr.Interface(
     fn=chatbot,
     inputs=["text", "state"],
     outputs=["chatbot", "state"],
-    title="Tu Compañero AI con Llama-2",
-    description="Un chatbot de IA avanzado utilizando el modelo Llama-2-7b-chat para conversaciones coherentes y naturales.",
+    title="Tu Compañero AI con OPT-1.3B",
+    description="Un chatbot de IA utilizando el modelo OPT-1.3B para conversaciones coherentes y naturales.",
 )
 
 iface.launch()