Hugging Face's logo

mpd
/
test_tea

Image Classification
fastai
Model card Files Community
mpd commited on
Commit
55cecd4
·
1 Parent(s): b5d45b6

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
 
8
  print('PIPELINE')
9
 
10
- class PreTrainedPipeline:
11
  def __init__(self, path=""):
12
  # IMPLEMENT_THIS
13
  # Preload all the elements you are going to need at inference.
 
7
 
8
  print('PIPELINE')
9
 
10
+ class ImageClassificationPipeline:
11
  def __init__(self, path=""):
12
  # IMPLEMENT_THIS
13
  # Preload all the elements you are going to need at inference.