HoneyTian commited on
Commit
fa5f1d9
·
1 Parent(s): b89d471
Files changed (1) hide show
  1. main.py +12 -2
main.py CHANGED
@@ -43,6 +43,11 @@ def get_args():
43
  default=(project_path / "data/eval_data").as_posix(),
44
  type=str,
45
  )
 
 
 
 
 
46
  args = parser.parse_args()
47
  return args
48
 
@@ -107,7 +112,12 @@ def load_board():
107
 
108
  with open(filename.as_posix(), "r", encoding="utf-8") as f:
109
  for row in f:
110
- row = json.loads(row)
 
 
 
 
 
111
  if name.endswith("-choice"):
112
  score_ = row["correct"]
113
  elif name.endswith("-chat"):
@@ -313,7 +323,7 @@ def main():
313
  share=False if platform.system() == "Windows" else False,
314
  server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
315
  # server_name="0.0.0.0",
316
- server_port=environment.get("port", 7860, dtype=int),
317
  )
318
  return
319
 
 
43
  default=(project_path / "data/eval_data").as_posix(),
44
  type=str,
45
  )
46
+ parser.add_argument(
47
+ "--server_port",
48
+ default=7862,
49
+ type=int,
50
+ )
51
  args = parser.parse_args()
52
  return args
53
 
 
112
 
113
  with open(filename.as_posix(), "r", encoding="utf-8") as f:
114
  for row in f:
115
+ try:
116
+ row = json.loads(row)
117
+ except Exception as e:
118
+ print(f"json load row failed. error type: {type(e)}, error text: {str(e)}")
119
+ logger.error(f"json load row failed. error type: {type(e)}, error text: {str(e)}")
120
+ raise e
121
  if name.endswith("-choice"):
122
  score_ = row["correct"]
123
  elif name.endswith("-chat"):
 
323
  share=False if platform.system() == "Windows" else False,
324
  server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
325
  # server_name="0.0.0.0",
326
+ server_port=environment.get("port", default=args.server_port, dtype=int),
327
  )
328
  return
329