Spaces:

raja-7-c
/

Depression-Severity-Predictor-BERT

Runtime error

raja-7-c commited on May 21, 2023

Commit

66a0e22

1 Parent(s): b37ee5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ tokenizer = Tokenizer(num_words=vocab_size)
 tokenizer.fit_on_texts(all_participants_mix['answer'])
 tokenizer.fit_on_sequences(all_participants_mix['answer'])
-labels = ['None', 'None', 'Mild', 'Moderate', 'Severe']
 num_classes = len(labels)
 def plot_acc(history, title="Model Accuracy"):

 tokenizer.fit_on_texts(all_participants_mix['answer'])
 tokenizer.fit_on_sequences(all_participants_mix['answer'])
+labels = ['None', 'None', 'Mild', 'Moderate', 'Moderately Severe']
 num_classes = len(labels)
 def plot_acc(history, title="Model Accuracy"):