coldnasser commited on
Commit
b15f93d
·
1 Parent(s): be8cfaf

Use new v2 model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,13 +10,13 @@ label_mapping = {
10
  }
11
 
12
  # Load classifier
13
- classifier = pipeline("text-classification", model="coldnasser/depression-anxiety-mindscape")
14
 
15
  def predict(texts):
16
  try:
17
  if isinstance(texts, str):
18
  texts = [texts]
19
-
20
  results = classifier(texts)
21
 
22
  # Initialize score aggregator
 
10
  }
11
 
12
  # Load classifier
13
+ classifier = pipeline("text-classification", model="username/mindscape-v2")
14
 
15
  def predict(texts):
16
  try:
17
  if isinstance(texts, str):
18
  texts = [texts]
19
+
20
  results = classifier(texts)
21
 
22
  # Initialize score aggregator