da03 commited on
Commit
dde88b9
1 Parent(s): 9eda610
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def predict_product(num1, num2):
64
  model = models[model_name]
65
  if finished_per_model[model_name]:
66
  continue
67
- if step >= MAX_PRODUCT_DIGITS_PER_MODE[model_name]:
68
  continue
69
  generation_kwargs = {
70
  'input_ids': generated_ids_per_model[model_name],
 
64
  model = models[model_name]
65
  if finished_per_model[model_name]:
66
  continue
67
+ if step >= MAX_PRODUCT_DIGITS_PER_MODEL[model_name]:
68
  continue
69
  generation_kwargs = {
70
  'input_ids': generated_ids_per_model[model_name],