Update src/brain.py

src/brain.py

@@ -7,6 +7,6 @@ def generate_answers(query, data):
     inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)
     outputs = model(**inputs)
     prediction = torch.argmax(outputs.logits, dim=1)
-    labels = data.columns  # Substitua com suas etiquetas reais
+    labels = ['ds','real','Group']
     predicted_label = labels[prediction]
     return predicted_label