Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,7 +106,7 @@ def test(gpu_id, net, img_list, group_size, img_size):
|
|
| 106 |
print('done')
|
| 107 |
return result
|
| 108 |
|
| 109 |
-
img_lst=[(torch.rand(352,352,3)*255).numpy().astype(np.uint8) for i in range(
|
| 110 |
|
| 111 |
|
| 112 |
|
|
|
|
| 106 |
print('done')
|
| 107 |
return result
|
| 108 |
|
| 109 |
+
img_lst=[(torch.rand(352,352,3)*255).numpy().astype(np.uint8) for i in range(5)]
|
| 110 |
|
| 111 |
|
| 112 |
|