Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,9 @@ def process_image(image):
|
|
| 27 |
).squeeze()
|
| 28 |
output = prediction.cpu().numpy()
|
| 29 |
formatted = (output * 255 / np.max(output)).astype('uint8')
|
|
|
|
| 30 |
img = Image.fromarray(formatted)
|
| 31 |
-
return img
|
| 32 |
|
| 33 |
return result
|
| 34 |
|
|
|
|
| 27 |
).squeeze()
|
| 28 |
output = prediction.cpu().numpy()
|
| 29 |
formatted = (output * 255 / np.max(output)).astype('uint8')
|
| 30 |
+
|
| 31 |
img = Image.fromarray(formatted)
|
| 32 |
+
return [img, gltf_path, gltf_path]
|
| 33 |
|
| 34 |
return result
|
| 35 |
|