Bug fix

app.py

@@ -49,12 +49,10 @@ class Processor:
             num_return_sequences=m,
             top_p=top_p,
         ) # (K, KL); begins with 0 ([BOS]); ends with 1 ([EOS])
-        print(knowledges_ids)
         knowledges_ids = knowledges_ids[:, 1:].contiguous() # no beginning; ends with 1 ([EOS])
         knowledges = self.tokenizer.batch_decode(knowledges_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True)
         knowledges = list(set(knowledges))
         knowledges = [''] + knowledges
-        print(knowledges)
 
         prompts = [question + (f' \\n Knowledge: {knowledge} \\n Answer: ' if knowledge != '' else ' \\n Answer:') for knowledge in knowledges]
         prompts_tok = self.tokenizer(prompts, return_tensors='pt', padding='max_length', truncation='longest_first', max_length=max_question_len + max_knowledge_len).to(device) # (1+K, QL+KL)