Spaces:
Sleeping
Sleeping
update stuff
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ transform = transforms.Compose([
|
|
41 |
def predict(image):
|
42 |
image = image['composite']
|
43 |
image = Image.fromarray(image)
|
44 |
-
input = transform(image).
|
45 |
print(image.shape)
|
46 |
|
47 |
with torch.no_grad():
|
|
|
41 |
def predict(image):
|
42 |
image = image['composite']
|
43 |
image = Image.fromarray(image)
|
44 |
+
input = transform(image).unsqueeze(0)
|
45 |
print(image.shape)
|
46 |
|
47 |
with torch.no_grad():
|