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"
|