oskarastrom commited on
Commit
f7827d6
·
1 Parent(s): b39c376

Update pdf_handler.py

Browse files
Files changed (1) hide show
  1. gradio_scripts/pdf_handler.py +2 -1
gradio_scripts/pdf_handler.py CHANGED
@@ -286,7 +286,7 @@ def generate_fish_tracks(pdf, result, id):
286
  plt.text(h*(1-Y[0]), w*(1-X[0]), "Start", fontsize=15, weight="bold")
287
  plt.text(h*(1-Y[-1]), w*(1-X[-1]), "End", fontsize=15, weight="bold")
288
 
289
- colors = []
290
  for i in range(1, len(X)):
291
 
292
  certain = certainty[i]
@@ -304,6 +304,7 @@ def generate_fish_tracks(pdf, result, id):
304
 
305
  print(len(X))
306
  print(len(Y))
 
307
  for i in range(1, len(X)):
308
  ax.plot(h*(1-Y[i]), w*(1-X[i]), color=colors[i], marker='o', markersize=3)
309
 
 
286
  plt.text(h*(1-Y[0]), w*(1-X[0]), "Start", fontsize=15, weight="bold")
287
  plt.text(h*(1-Y[-1]), w*(1-X[-1]), "End", fontsize=15, weight="bold")
288
 
289
+ colors = [""]
290
  for i in range(1, len(X)):
291
 
292
  certain = certainty[i]
 
304
 
305
  print(len(X))
306
  print(len(Y))
307
+ print(len(colors))
308
  for i in range(1, len(X)):
309
  ax.plot(h*(1-Y[i]), w*(1-X[i]), color=colors[i], marker='o', markersize=3)
310