Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ def detect(image):
|
|
| 56 |
size = max(image.size)
|
| 57 |
new_im = Image.new('RGB', (size, size), color = 0) # Create a squared black image
|
| 58 |
new_im.paste(image)
|
| 59 |
-
|
| 60 |
|
| 61 |
with torch.no_grad():
|
| 62 |
|
|
|
|
| 56 |
size = max(image.size)
|
| 57 |
new_im = Image.new('RGB', (size, size), color = 0) # Create a squared black image
|
| 58 |
new_im.paste(image)
|
| 59 |
+
new_ im = Image.fromarray(new_im.astype(np.uint8))
|
| 60 |
|
| 61 |
with torch.no_grad():
|
| 62 |
|