axiilay commited on
Commit
94fd0fd
·
1 Parent(s): ca8cbba

enable eval_mode to make infer return value

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -70,6 +70,7 @@ def process_image(image, model_size, task_type):
70
  crop_mode=config["crop_mode"],
71
  save_results=True,
72
  test_compress=True,
 
73
  )
74
 
75
  print(f"====\nresult: {result}\n====\n")
 
70
  crop_mode=config["crop_mode"],
71
  save_results=True,
72
  test_compress=True,
73
+ eval_mode=True,
74
  )
75
 
76
  print(f"====\nresult: {result}\n====\n")