Update README.md
Browse files
README.md
CHANGED
|
@@ -32,11 +32,11 @@ import numpy as np
|
|
| 32 |
import joblib
|
| 33 |
import requests
|
| 34 |
|
| 35 |
-
model_path = "
|
| 36 |
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
| 37 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 38 |
|
| 39 |
-
mlb_url = "https://huggingface.co/
|
| 40 |
mlb_path = "mlb.pkl"
|
| 41 |
|
| 42 |
with open(mlb_path, "wb") as f:
|
|
|
|
| 32 |
import joblib
|
| 33 |
import requests
|
| 34 |
|
| 35 |
+
model_path = "vtt-qsts-ai/multilabel-indicator-classification-roberta-base"
|
| 36 |
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
| 37 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 38 |
|
| 39 |
+
mlb_url = "https://huggingface.co/vtt-qsts-ai/multilabel-indicator-classification-roberta-base/resolve/main/mlb.pkl"
|
| 40 |
mlb_path = "mlb.pkl"
|
| 41 |
|
| 42 |
with open(mlb_path, "wb") as f:
|