Update app.py

app.py

@@ -15,7 +15,7 @@ def classify_medical_text(txt):
     return dict(zip(medical_categories, map(float, probs)))
 
 # Load the psychiatric model from Hugging Face
-psychiatric_model_name = "mental/mental-bert-base-uncased"  # Replace with the appropriate model
+psychiatric_model_name = "nlp4good/psych-search"  # Replace with the appropriate model
 psychiatric_tokenizer = AutoTokenizer.from_pretrained(psychiatric_model_name)
 psychiatric_model = AutoModelForSequenceClassification.from_pretrained(psychiatric_model_name)