Jiahuita
commited on
Commit
·
bb8ddb0
1
Parent(s):
acdeca7
modified error in pipeline
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -6,7 +6,7 @@ import numpy as np
|
|
| 6 |
import json
|
| 7 |
|
| 8 |
class NewsClassifierConfig(PretrainedConfig):
|
| 9 |
-
model_type = "
|
| 10 |
|
| 11 |
def __init__(
|
| 12 |
self,
|
|
|
|
| 6 |
import json
|
| 7 |
|
| 8 |
class NewsClassifierConfig(PretrainedConfig):
|
| 9 |
+
model_type = "text_classifier"
|
| 10 |
|
| 11 |
def __init__(
|
| 12 |
self,
|