Update generate_answer.py

generate_answer.py

@@ -21,5 +21,5 @@ def generate_answer(question):
     output = model.generate(**encoded_input, max_length=100)  # Use generate method for text generation
     # Decode the model output to text
     decoded_output = tokenizer.decode(output[0])
-    response_text = decoded_output.split('\n\n', 1)
-    return response_text[1]
+    # response_text = decoded_output.split('\n\n', 1)
+    return decoded_output