Rolv-Arild commited on
Commit
9a6da2e
·
verified ·
1 Parent(s): 67001fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -273,10 +273,10 @@ def gradio_app():
273
  postfix += "_nullify_goal_difference"
274
  elif ignore_ties:
275
  postfix += "_ignore_ties"
276
- preds_file = os.path.join(temp_dir, f"predictions_{replay_stem}_{postfix}.csv")
277
  if os.path.exists(preds_file):
278
  print(f"Predictions file already exists: {preds_file}")
279
- preds = pd.read_csv(preds_file)
280
  else:
281
  preds = infer(MODEL, replay_file,
282
  nullify_goal_difference=nullify_goal_difference,
 
273
  postfix += "_nullify_goal_difference"
274
  elif ignore_ties:
275
  postfix += "_ignore_ties"
276
+ preds_file = os.path.join(temp_dir, f"predictions_{replay_stem}{postfix}.csv")
277
  if os.path.exists(preds_file):
278
  print(f"Predictions file already exists: {preds_file}")
279
+ preds = pd.read_csv(preds_file, dtype={"Touch": str})
280
  else:
281
  preds = infer(MODEL, replay_file,
282
  nullify_goal_difference=nullify_goal_difference,