Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastai.text.all import *
-from transformers import BlenderbotForConditionalGeneration, BlenderbotTokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, BlenderbotForConditionalGeneration, BlenderbotTokenizer
 import torch
 import gradio as gr
 
@@ -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 = "nlp4good/psych-search"  # Replace with the appropriate model
+psychiatric_model_name = "mental/mental-bert-base-uncased"  # Replace with the appropriate model
 psychiatric_tokenizer = AutoTokenizer.from_pretrained(psychiatric_model_name)
 psychiatric_model = AutoModelForSequenceClassification.from_pretrained(psychiatric_model_name)
 
@@ -96,8 +96,4 @@ with gr.Blocks() as lifestyle_interface:
 
 # Combine interfaces using Tabs
 app = gr.TabbedInterface(
-    [medical_interface, psychiatric_interface, lifestyle_interface],
-    ["Medical Diagnosis", "Psychiatric Analysis", "Lifestyle & Nutrition Chat"]
-)
-
-app.launch(inline=False)
+    [medi