Commit
•
8899223
1
Parent(s):
6363c74
Update pipeline.py
Browse files- 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(
|
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 |
|