Spaces:
Runtime error
Runtime error
Commit
•
eef8577
1
Parent(s):
7656d23
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,6 +28,8 @@ def visualize_model_output(prediction):
|
|
| 28 |
|
| 29 |
output = np.zeros(prediction.shape)
|
| 30 |
|
|
|
|
|
|
|
| 31 |
for unq_class in unique_classes:
|
| 32 |
rgb_class_unique = rgb_colors[str(unq_class)]
|
| 33 |
output[:,:,0] = rgb_class_unique[0]
|
|
|
|
| 28 |
|
| 29 |
output = np.zeros(prediction.shape)
|
| 30 |
|
| 31 |
+
print(output.shape,'shape of output')
|
| 32 |
+
|
| 33 |
for unq_class in unique_classes:
|
| 34 |
rgb_class_unique = rgb_colors[str(unq_class)]
|
| 35 |
output[:,:,0] = rgb_class_unique[0]
|