frostymelonade commited on
Commit
695c722
Β·
1 Parent(s): 22d9c37

dummy commit (removed commented code)

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,17 +1,15 @@
1
  from transformers import AutoTokenizer
2
  from transformers import AutoModelForSequenceClassification
3
  from transformers import DataCollatorWithPadding
4
- from transformers import TrainingArguments, Trainer
5
  import gradio as gr
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("smallbenchnlp/roberta-small")
8
- #training_args = TrainingArguments(output_dir="roberta-small-pun-detector", evaluation_strategy="epoch")
9
  model = AutoModelForSequenceClassification.from_pretrained("frostymelonade/roberta-small-pun-detector-v2", num_labels=2)
10
  data_collator = DataCollatorWithPadding(tokenizer=tokenizer)
11
 
12
  trainer = Trainer(
13
  model=model,
14
- # args=training_args,
15
  data_collator=data_collator,
16
  tokenizer=tokenizer,
17
  )
 
1
  from transformers import AutoTokenizer
2
  from transformers import AutoModelForSequenceClassification
3
  from transformers import DataCollatorWithPadding
4
+ from transformers import Trainer
5
  import gradio as gr
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("smallbenchnlp/roberta-small")
 
8
  model = AutoModelForSequenceClassification.from_pretrained("frostymelonade/roberta-small-pun-detector-v2", num_labels=2)
9
  data_collator = DataCollatorWithPadding(tokenizer=tokenizer)
10
 
11
  trainer = Trainer(
12
  model=model,
 
13
  data_collator=data_collator,
14
  tokenizer=tokenizer,
15
  )