Fix
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ def predict(patch, max_length, min_length, num_beams, prediction_count):
|
|
68 |
)
|
69 |
|
70 |
result = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
71 |
-
return token_count,
|
72 |
|
73 |
|
74 |
iface = gr.Interface(fn=predict, inputs=[
|
|
|
68 |
)
|
69 |
|
70 |
result = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
71 |
+
return token_count, input_text, {k: 0 for k in result}
|
72 |
|
73 |
|
74 |
iface = gr.Interface(fn=predict, inputs=[
|