Update app.py

app.py

@@ -12,8 +12,8 @@ import os
 #base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
 
 sft_model = "somosnlp/RecetasDeLaAbuela5k_gemma-2b-bnb-4bit"
-#base_model_name = "unsloth/gemma-2b-bnb-4bit"
-base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
+base_model_name = "unsloth/gemma-2b-bnb-4bit"
+#base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
 
 max_seq_length=300
 base_model = AutoModelForCausalLM.from_pretrained(base_model_name,return_dict=True,device_map="auto", torch_dtype=torch.float16,)