Commit
·
53b155c
1
Parent(s):
11d3b20
refactor: update 500 model
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ token = os.environ.get("HF_TOKEN")
|
|
9 |
raw_model_name = "distilbert-base-uncased"
|
10 |
raw_model = pipeline("sentiment-analysis", model=raw_model_name)
|
11 |
|
12 |
-
scademy_500_500_model_name = "scademy/
|
13 |
scademy_500_500_model = pipeline("sentiment-analysis", model=scademy_500_500_model_name, token=token)
|
14 |
|
15 |
scademy_2500_2500_model_name = "scademy/bert-finetuned-2500-2500-0"
|
|
|
9 |
raw_model_name = "distilbert-base-uncased"
|
10 |
raw_model = pipeline("sentiment-analysis", model=raw_model_name)
|
11 |
|
12 |
+
scademy_500_500_model_name = "scademy/DistilBert-500-500-0"
|
13 |
scademy_500_500_model = pipeline("sentiment-analysis", model=scademy_500_500_model_name, token=token)
|
14 |
|
15 |
scademy_2500_2500_model_name = "scademy/bert-finetuned-2500-2500-0"
|