QSTS-VTT commited on
Commit
b3dd52a
·
verified ·
1 Parent(s): 068612e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -32,11 +32,11 @@ import numpy as np
32
  import joblib
33
  import requests
34
 
35
- model_path = "toqeerehsan/multilabel-indicator-classification"
36
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
37
  tokenizer = AutoTokenizer.from_pretrained(model_path)
38
 
39
- mlb_url = "https://huggingface.co/toqeerehsan/multilabel-indicator-classification/resolve/main/mlb.pkl"
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: