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")
|