Alexandre-Numind commited on
Commit
1a7a373
·
verified ·
1 Parent(s): 111fbee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -145,7 +145,7 @@ def gradio_interface_function(template, text, size, is_example):
145
  return # End the function since there was an error
146
 
147
  # Initialize the sliding window prediction process
148
- prediction_generator = sliding_window_prediction(template, text, model, tokenizer, window_size=size)
149
 
150
  # Iterate over the generator to return values at each step
151
  for progress, full_pred, html_content in prediction_generator:
 
145
  return # End the function since there was an error
146
 
147
  # Initialize the sliding window prediction process
148
+ prediction_generator = sliding_window_prediction(template, text, model, tokenizer, window_size=int(size))
149
 
150
  # Iterate over the generator to return values at each step
151
  for progress, full_pred, html_content in prediction_generator: