Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,6 @@ import textwrap
|
|
6 |
# Define the model to be used
|
7 |
#model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
8 |
|
9 |
-
|
10 |
# Load model directly
|
11 |
from transformers import AutoModel
|
12 |
model = AutoModel.from_pretrained("GRMenon/mental-health-mistral-7b-instructv0.2-finetuned-V2")
|
|
|
6 |
# Define the model to be used
|
7 |
#model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
8 |
|
|
|
9 |
# Load model directly
|
10 |
from transformers import AutoModel
|
11 |
model = AutoModel.from_pretrained("GRMenon/mental-health-mistral-7b-instructv0.2-finetuned-V2")
|