yeliudev commited on
Commit
b5c1599
·
verified ·
1 Parent(s): d11572f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -137,8 +137,8 @@ def main(video, prompt, role, temperature, max_new_tokens, model, processor, str
137
  gr.Warning('Verifier cannot be used without Grounder.')
138
  return history
139
 
140
- if 'pla' in role and any(k not in role for k in ('gnd', 'ver', 'ans')):
141
- gr.Warning('Planner can only be used when all other roles are selected.')
142
  return history
143
 
144
  history.append({'role': 'user', 'content': prompt})
 
137
  gr.Warning('Verifier cannot be used without Grounder.')
138
  return history
139
 
140
+ if 'pla' in role and 'gnd' not in role and 'ver' not in role:
141
+ gr.Warning('Planner can only be used with Grounder and Verifier.')
142
  return history
143
 
144
  history.append({'role': 'user', 'content': prompt})