Lord-Raven commited on
Commit
4b7b387
·
1 Parent(s): 07c25e9

Experimenting with few-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ args = TrainingArguments(
80
  num_epochs=1
81
  )
82
  trainer = Trainer(
83
- model=ort_model,
84
  args=args,
85
  train_dataset=train_dataset,
86
  eval_dataset=reference_dataset["test"]
 
80
  num_epochs=1
81
  )
82
  trainer = Trainer(
83
+ model=few_shot_model,
84
  args=args,
85
  train_dataset=train_dataset,
86
  eval_dataset=reference_dataset["test"]