orisuchy's picture
Update README.md
5d8fe1a
metadata
license: afl-3.0
language: he
tags:
  - Text Classification
widget:
  - text: היער השחור והגדול
  - text: ואז הוא הלך לטייל בתוך היער השחור והגדול
datasets:
  - orisuchy/Descriptive_Sentences_He
metrics:
  - accuracy

Descriptive Sentences Classifier

Based on AlephBERT model.

How to Use the model:

from transformers import pipeline
classifier = pipeline("text-classification",model='orisuchy/Descriptive_Classifier', return_all_scores=True)
outputs = classifier("מסווג חתיך במיוחד")
print(outputs)

"""
Output:
[[
{'label': 'Descriptive', 'score': 0.9998779296875},
{'label': 'Not Descriptive', 'score': 0.0001220278863911517}]]
"""

Or, if you want only the final class:

from transformers import pipeline
classifier = pipeline("text-classification",model='orisuchy/Descriptive_Classifier')
output = classifier("הלכתי אליו הביתה וחיכיתי")
print(output)

"""
Output:
[{'label': 'Not Descriptive', 'score': 0.9999438524246216}]
"""

Created by Daniel Smotritsky & Ori Suchy