Update app.py

app.py

@@ -57,7 +57,7 @@ def generate_answer(query):
     ids = generator.generate(inputs["input_ids"], num_beams=2, min_length=20, max_length=64)
     # use tokenizer to decode the output ids
     answer = tokenizer.batch_decode(ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
-    return pprint(answer)
+    st.write(str(answer))
 
 query = st.text_area('Enter Question:')
 b = st.button('Submit!')