Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
|
|
14 |
def summarize_text(input_text):
|
15 |
summar_input = 'summarize: '+input_text
|
16 |
input_tokens = tokenizer(summar_input, return_tensors='pt').input_ids
|
17 |
-
outputs = model.generate(input_tokens)
|
18 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
19 |
|
20 |
# Interface for the Gradio app
|
|
|
14 |
def summarize_text(input_text):
|
15 |
summar_input = 'summarize: '+input_text
|
16 |
input_tokens = tokenizer(summar_input, return_tensors='pt').input_ids
|
17 |
+
outputs = model.generate(input_tokens, max_new_tokens=100, min_new_tokens=30, do_sample=False)
|
18 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
19 |
|
20 |
# Interface for the Gradio app
|