Update app.py

app.py

@@ -11,8 +11,11 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import os
 
 #sft_model = "somosnlp/gemma-FULL-RAC-Colombia_v2"
-sft_model = "somosnlp/RecetasDeLaAbuela_mistral-7b-instruct-v0.2-bnb-4bit"
-base_model_name = "mistralai/Mistral-7B-Instruct-v0.2"
+#sft_model = "somosnlp/RecetasDeLaAbuela_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"
+base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
+
 
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,