DD0101 commited on
Commit
0f2eb8c
1 Parent(s): df81629

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ class MyPipeline(TokenClassificationPipeline):
45
 
46
  return model_inputs
47
 
48
- model_checkpoint = "DD0101/disfluency-base"
49
 
50
  my_classifier = pipeline(
51
  "token-classification", model=model_checkpoint, aggregation_strategy="simple", pipeline_class=MyPipeline)
 
45
 
46
  return model_inputs
47
 
48
+ model_checkpoint = "DD0101/disfluency-large"
49
 
50
  my_classifier = pipeline(
51
  "token-classification", model=model_checkpoint, aggregation_strategy="simple", pipeline_class=MyPipeline)