update

pages/summarization_playground.py

@@ -107,14 +107,14 @@ def get_model_batch_generation(model_name):
     return model[model_name]
 
 
-def generate_answer(sources, model_name, prompt, temperature, max_new_tokens, do_sample):
+def generate_answer(sources, model_name, prompt, temperature=0.0001, max_new_tokens=500, do_sample=True):
     model_device_check(model_name)
     content = prompt + '\n{' + sources + '}\n\nsummary:'
     answer = model[model_name].gen(content,temperature,max_new_tokens,do_sample)[0].strip()
 
     return answer
 
-def process_input(input_text, model_selection, prompt, temperature, max_new_tokens, do_sample):
+def process_input(input_text, model_selection, prompt, temperature=0.0001, max_new_tokens=500, do_sample=True):
     if input_text:
         logging.info("Start generation")
         response = generate_answer(input_text, model_selection, prompt, temperature, max_new_tokens, do_sample)