Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
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:
|