Update app.py
Browse files
app.py
CHANGED
|
@@ -41,6 +41,16 @@ def preprocess_text(input_text, tokenizer):
|
|
| 41 |
padded_sequences = np.array(sequences) # Convert list to NumPy array
|
| 42 |
return padded_sequences
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
def main():
|
| 45 |
# Text input for mental state
|
| 46 |
input_text = st.text_input("Enter the Mental state here...")
|
|
|
|
| 41 |
padded_sequences = np.array(sequences) # Convert list to NumPy array
|
| 42 |
return padded_sequences
|
| 43 |
|
| 44 |
+
class_labels = {
|
| 45 |
+
1: "Normal",
|
| 46 |
+
2: "Depression",
|
| 47 |
+
3: "Suicidal",
|
| 48 |
+
4: "Anxiety",
|
| 49 |
+
5: "Bipolar",
|
| 50 |
+
6: "Stress",
|
| 51 |
+
7: "Personality disorder"
|
| 52 |
+
}
|
| 53 |
+
|
| 54 |
def main():
|
| 55 |
# Text input for mental state
|
| 56 |
input_text = st.text_input("Enter the Mental state here...")
|