fastai
/

fastbook_04_mnist_basics

Image Classification

Model card Files Files and versions Community

espejelomar commited on May 18, 2022

Commit

633dc2b

•

1 Parent(s): 5570344

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -5,7 +5,7 @@ from huggingface_hub import from_pretrained_fastai
 from PIL import Image
-class ImageClassificationPipeline(Pipeline):
     def __init__(self, model_id: str):
         self.model = from_pretrained_fastai(model_id)

 from PIL import Image
+class ImageClassificationPipeline():
     def __init__(self, model_id: str):
         self.model = from_pretrained_fastai(model_id)