djl234 commited on
Commit
963faaf
·
1 Parent(s): 5bcd599

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(100000)]
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