Update app.py

app.py

@@ -66,10 +66,10 @@ tokenizer.padding_side = 'left'
 # Load the GaiaMiniMed model with the specified configuration
 # Load the Peft model with a specific configuration
 # Specify the configuration class for the model
-model_config = PeftConfig.from_pretrained(base_model_id)
+model_config = AutoConfig.from_pretrained(base_model_id)
 # Load the PEFT model with the specified configuration
-peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
-peft_model = PeftModel.from_pretrained(model=base_model_id, model_id=model_directory)
+peft_model = AutoModelForCausalLM.from_pretrained(model_directory, config=model_config)
+peft_model = PeftModel.from_pretrained(model=base_model_id,, trust_remote_code=True)
 peft_model = PeftModel.from_pretrained(peft_model, model_directory)