davanstrien HF staff commited on
Commit
8899223
1 Parent(s): 6363c74

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -49,8 +49,8 @@ class PreTrainedPipeline():
49
  # IMPLEMENT_THIS
50
 
51
  # FastAI expects a np array, not a PIL Image.
52
-
53
- _, _, preds = self.model.predict(np.array(inputs))
54
 
55
  preds = preds.tolist()
56
 
 
49
  # IMPLEMENT_THIS
50
 
51
  # FastAI expects a np array, not a PIL Image.
52
+
53
+ _, _, preds = self.model.predict(inputs)
54
 
55
  preds = preds.tolist()
56