kairaamilanii
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,6 @@ metrics:
|
|
11 |
- f1
|
12 |
base_model:
|
13 |
- FacebookAI/roberta-base
|
14 |
-
new_version: kairaamilanii/RoBERTa-minds-14-classifier-intent
|
15 |
pipeline_tag: text-classification
|
16 |
model-index:
|
17 |
- name: roBERTa-minds14-en-classifier
|
@@ -55,3 +54,12 @@ model-index:
|
|
55 |
---
|
56 |
|
57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
- f1
|
12 |
base_model:
|
13 |
- FacebookAI/roberta-base
|
|
|
14 |
pipeline_tag: text-classification
|
15 |
model-index:
|
16 |
- name: roBERTa-minds14-en-classifier
|
|
|
54 |
---
|
55 |
|
56 |
|
57 |
+
from transformers import pipeline
|
58 |
+
|
59 |
+
model = "/content/RoBERTa-mind14-classifier-intent"
|
60 |
+
classifier = pipeline("text-classification", model=model)
|
61 |
+
|
62 |
+
text = "hi what's the maximum amount of money I can withdraw from" # Replace with your desired input text
|
63 |
+
|
64 |
+
prediction = classifier(text)
|
65 |
+
prediction
|