Spaces:
Runtime error
Runtime error
Hector Lopez
commited on
Commit
·
26675f2
1
Parent(s):
f7cb0f8
Fix
Browse files
model.py
CHANGED
|
@@ -37,6 +37,7 @@ def get_checkpoint(checkpoint_path):
|
|
| 37 |
|
| 38 |
def predict(model, image):
|
| 39 |
img = PIL.Image.open(image)
|
|
|
|
| 40 |
class_map = ClassMap(classes=['Waste'])
|
| 41 |
transforms = tfms.A.Adapter([
|
| 42 |
*tfms.A.resize_and_pad(512),
|
|
|
|
| 37 |
|
| 38 |
def predict(model, image):
|
| 39 |
img = PIL.Image.open(image)
|
| 40 |
+
print(img.shape)
|
| 41 |
class_map = ClassMap(classes=['Waste'])
|
| 42 |
transforms = tfms.A.Adapter([
|
| 43 |
*tfms.A.resize_and_pad(512),
|