Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
4b7b387
1
Parent(s):
07c25e9
Experimenting with few-shot classification.
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ args = TrainingArguments(
|
|
80 |
num_epochs=1
|
81 |
)
|
82 |
trainer = Trainer(
|
83 |
-
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"]
|