nickmuchi commited on
Commit
888c0d2
·
1 Parent(s): f2d241e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
47
  if id2label is not None:
48
  labels = [id2label[x] for x in labels]
49
 
50
- plt.figure(figsize=(18, 13))
51
  plt.imshow(img)
52
  ax = plt.gca()
53
  colors = COLORS * 100
 
47
  if id2label is not None:
48
  labels = [id2label[x] for x in labels]
49
 
50
+ plt.figure(figsize=(25, 20))
51
  plt.imshow(img)
52
  ax = plt.gca()
53
  colors = COLORS * 100