Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
-base_model = "heliosbrahma/falcon-7b-sharded-bf16-finetuned-mental-health-conversational"
+base_model = "ybelkada/falcon-7b-sharded-bf16"
 adapter_model = "Sanjay002/falcon-7b-mental-health-finetuned"
 
 tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)