Bug fix

app.py

@@ -55,7 +55,7 @@ class Processor:
         knowledges = [''] + 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).input_ids.to(device) # (1+K, QL+KL)
+        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)
         output = self.model(
             input_ids=prompts_tok.input_ids,
             attention_mask=prompts_tok.attention_mask,