acecalisto3 commited on
Commit
fb8f864
·
verified ·
1 Parent(s): 70e24c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,6 @@ import os
2
  from typing import Dict, List, Optional
3
  from dataclasses import dataclass, field
4
  from transformers import (
5
- AutoConfig,
6
  AutoModelForSequenceClassification,
7
  AutoTokenizer,
8
  DataCollatorWithPadding,
@@ -11,6 +10,8 @@ from transformers import (
11
  TrainingArguments,
12
  )
13
 
 
 
14
  @dataclass
15
  class ModelArguments:
16
  model_name: str = field(
 
2
  from typing import Dict, List, Optional
3
  from dataclasses import dataclass, field
4
  from transformers import (
 
5
  AutoModelForSequenceClassification,
6
  AutoTokenizer,
7
  DataCollatorWithPadding,
 
10
  TrainingArguments,
11
  )
12
 
13
+ from transformers.models import AutoConfig
14
+
15
  @dataclass
16
  class ModelArguments:
17
  model_name: str = field(