Gustking commited on
Commit
c4d8a1d
·
1 Parent(s): e1464f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,6 +63,7 @@ def main():
63
  return retornos_btn
64
 
65
  def set_col_visible(board,game_over,temp,usr,ia,selected):
 
66
  usr = 1 if selected == 'X' else 2
67
  ia = 2 if usr == 1 else 1
68
  retorn_buttons = [gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=True)]*9
@@ -107,7 +108,7 @@ def main():
107
  return gr.update(value=msg, show_label=False), gr.Number(value=int(game_over), visible=False)
108
 
109
  parser = argparse.ArgumentParser()
110
- parser.add_argument('--file_name', type=str, required=False, default='models/q_agent-0.9ep-0.6ap-0.9-1000r.json')
111
  args = parser.parse_args()
112
  q_agent= QlearningAgent(epsilon=0,alpha=0,discount_factor=0, train=False)
113
  q_agent.load_agent_dict(args.file_name)
 
63
  return retornos_btn
64
 
65
  def set_col_visible(board,game_over,temp,usr,ia,selected):
66
+ game_over = -1
67
  usr = 1 if selected == 'X' else 2
68
  ia = 2 if usr == 1 else 1
69
  retorn_buttons = [gr.Button(size='lg', scale = 0, min_width = 100,value='',elem_classes='gradio-button',interactive=True)]*9
 
108
  return gr.update(value=msg, show_label=False), gr.Number(value=int(game_over), visible=False)
109
 
110
  parser = argparse.ArgumentParser()
111
+ parser.add_argument('--file_name', type=str, required=False, default='models/q_agent-0.5ep-0.6ap-0.9-1000000r.json')
112
  args = parser.parse_args()
113
  q_agent= QlearningAgent(epsilon=0,alpha=0,discount_factor=0, train=False)
114
  q_agent.load_agent_dict(args.file_name)