Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,6 +34,7 @@ def greet(image):
|
|
| 34 |
print(image.min(), image.max())
|
| 35 |
image = image/255.0
|
| 36 |
image = image.unsqueeze(0)
|
|
|
|
| 37 |
image = torch.permute(image, [0,2,3,1])
|
| 38 |
out = model(image)
|
| 39 |
|
|
|
|
| 34 |
print(image.min(), image.max())
|
| 35 |
image = image/255.0
|
| 36 |
image = image.unsqueeze(0)
|
| 37 |
+
print(image.shape)
|
| 38 |
image = torch.permute(image, [0,2,3,1])
|
| 39 |
out = model(image)
|
| 40 |
|