Spaces:

nebiyu29
/

classification-2

Sleeping

classification-2 / app.py

Create app.py

5dd1abe 8 months ago

No virus

489 Bytes

	import streamlit as st
	from transformers import pipeline

	classifier = pipeline("zero-shot-classification", model="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli")

	sequence_to_classify = st.text_input("input the cases here")

	candidate_labels = ["depression", "anxiety", "bipolar disorder", "schizophrenia", "PTSD", "OCD", "ADHD", "autism", "eating disorder", "personality disorder", "phobia"]

	output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
	st.write(output)