LutherYTT commited on
Commit
31c7300
·
1 Parent(s): 44416eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ torch.cuda.empty_cache()
11
 
12
  model_name = "hfl/chinese-roberta-wwm-ext"
13
 
14
- class MultiTasRokBert(BertPreTrainedModel):
15
  def __init__(self, config, model_name_or_path):
16
  super().__init__(config)
17
  # Load backbone with pretrained weights if desired
 
11
 
12
  model_name = "hfl/chinese-roberta-wwm-ext"
13
 
14
+ class MultiTaskRoBert(BertPreTrainedModel):
15
  def __init__(self, config, model_name_or_path):
16
  super().__init__(config)
17
  # Load backbone with pretrained weights if desired