Update app.py

app.py

@@ -133,8 +133,6 @@ class RAG:
 
       input_text = "answer: " + " ".join(context) + " " + question
 
-      print(input_text)
-
       inputs = self.generator_tokenizer.encode(input_text, return_tensors='pt', max_length=1024, truncation=True).to(device)
       outputs = self.generator_model.generate(inputs, max_length=150, min_length=2, length_penalty=2.0, num_beams=4, early_stopping=True)
 
@@ -159,8 +157,6 @@ class RAG:
       if answer_end_index < answer_start_index:
         answer_start_index, answer_end_index = answer_end_index, answer_start_index
 
-      print(answer_start_index, answer_end_index)
-
       predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
       answer = self.generator_tokenizer.decode(predict_answer_tokens, skip_special_tokens=True)
       answer = answer.replace('\n', ' ').strip()