mental-disorder-classification

Running

kingabzpro commited on Jul 30

Commit

40c4a66

•

1 Parent(s): a34a5fe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def prediction(news):
     device_map="cpu",)
     outputs = pipe(prompt, max_new_tokens=2, do_sample=True, temperature=0.1)
-    preds = outputs[0]["generated_text"].split("label: ")[-1].strip())
     return preds

     device_map="cpu",)
     outputs = pipe(prompt, max_new_tokens=2, do_sample=True, temperature=0.1)
+    preds = outputs[0]["generated_text"].split("label: ")[-1].strip()
     return preds