Update custom_pipeline.py
Browse files- custom_pipeline.py +1 -8
custom_pipeline.py
CHANGED
@@ -42,11 +42,4 @@ PIPELINE_REGISTRY.register_pipeline(
|
|
42 |
pt_model=AutoModelForSequenceClassification,
|
43 |
default={"pt": ("hf-internal-testing/tiny-random-bert", "main")},
|
44 |
type="text", # current support type: text, audio, image, multimodal
|
45 |
-
)
|
46 |
-
|
47 |
-
pipe = pipeline("new-task")
|
48 |
-
print(pipe("This is a test"))
|
49 |
-
|
50 |
-
repo = Repository("test-dynamic-pipeline", clone_from="lysandre/test-dynamic-pipeline")
|
51 |
-
pipe.save_pretrained("test-dynamic-pipeline")
|
52 |
-
repo.push_to_hub()
|
|
|
42 |
pt_model=AutoModelForSequenceClassification,
|
43 |
default={"pt": ("hf-internal-testing/tiny-random-bert", "main")},
|
44 |
type="text", # current support type: text, audio, image, multimodal
|
45 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|