richardkovacs commited on
Commit
53b155c
·
1 Parent(s): 11d3b20

refactor: update 500 model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/bert-finetuned-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"
 
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"