Update src/brain.py

src/brain.py

@@ -3,7 +3,7 @@ from transformers import BertTokenizer, BertForSequenceClassification
 tokenizer = BertTokenizer.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
 model = BertForSequenceClassification.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
 
-def generate_answers(query, data):
+def generate_answers(query):
     inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)
     outputs = model(**inputs)
     prediction = torch.argmax(outputs.logits, dim=1)