JiheonJeong commited on
Commit
e8463e1
·
1 Parent(s): 04e6419

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -64,6 +64,10 @@ def random_sample():
64
  def reset_sample(*agents):
65
  image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
66
  new_agents = []
 
 
 
 
67
  for k in range(TOTAL_K):
68
  new_agents.append('')
69
  return image, *[gr.Textbox.update(value=answer) for answer in new_agents]
 
64
  def reset_sample(*agents):
65
  image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
66
  new_agents = []
67
+ global CURRENT_POSITION
68
+ global SAMPLE_IDX
69
+ CURRENT_POSITION = []
70
+ SAMPLE_IDX = []
71
  for k in range(TOTAL_K):
72
  new_agents.append('')
73
  return image, *[gr.Textbox.update(value=answer) for answer in new_agents]