Update app.py

app.py

@@ -22,7 +22,7 @@ adapter_id = "Prajjwalng/gemma_customercare_adapters" #adapter model
 
 # Initialize model and tokenizer (load only once)
 @st.cache_resource
-def load_model(model_id, adapter_id):
+def load_model(model_id):
     base_model = AutoModelForCausalLM.from_pretrained(
         model_id,
         low_cpu_mem_usage=True,
@@ -31,9 +31,9 @@ def load_model(model_id, adapter_id):
         device_map={"": 0} if torch.cuda.is_available() else "cpu"
     )
 
-    merged_model = PeftModel.from_pretrained(base_model, adapter_id)
+    #merged_model = PeftModel.from_pretrained(base_model, adapter_id)
     tokenizer = AutoTokenizer.from_pretrained(model_id, add_eos_token=True)
-    return merged_model, tokenizer
+    return base_model, tokenizer
 
 merged_model, tokenizer = load_model(model_id, adapter_id)