Update app.py

app.py

@@ -6,9 +6,9 @@ import json
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, GemmaTokenizer,  StoppingCriteria, StoppingCriteriaList, GenerationConfig
 import os
 
-#sft_model = "somosnlp/RecetasDeLaAbuela_mistral-7b-instruct-v0.2-bnb-4bit"
+#sft_model = "somosnlp/ComeBien_mistral-7b-instruct-v0.2-bnb-4bit"
 #base_model_name = "unsloth/Mistral-7B-Instruct-v0.2"
-sft_model = "somosnlp/RecetasDeLaAbuela_gemma-2b-it-bnb-4bit"
+sft_model = "somosnlp/ComeBien_gemma-2b-it-bnb-4bit"
 base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
 
 max_seq_length=400
@@ -71,16 +71,16 @@ def mostrar_respuesta(pregunta, contexto):
 
 # Ejemplos de preguntas
 mis_ejemplos = [
-    ["¿Dime la receta de la tortilla de patatatas?", "Cocinero español"],
-    ["¿Dime la receta del ceviche?", "Cocinero peruano"],
-    ["¿Como se cocinan unos autenticos frijoles?", "Cocinero de México"],
+    ["¿Dime el valor nutricional de la tortilla de patatatas?", "Cocinero español"],
+    ["¿Dime el valor nutricional del ceviche?", "Cocinero peruano"],
+    ["¿Dime el valor nutricional de los frijoles?", "Cocinero de México"],
 ]
 
 iface = gr.Interface(
     fn=mostrar_respuesta,
     inputs=[gr.Textbox(label="Pregunta"), gr.Textbox(label="Contexto", value="You are a helpful AI assistant. Eres un experto cocinero hispanoamericano."),],
     outputs=[gr.Textbox(label="Respuesta", lines=2),],
-    title="Recetas de la Abuel@",
+    title="ComeBien",
     description="Introduce tu pregunta sobre recetas de cocina.",
     examples=mis_ejemplos,
 )