Added answers ids

utils.py

@@ -38,4 +38,12 @@ class Test:
     
     def get_image_path(self, idx):
         image_path = f"images/{idx}.jpg"
-        return image_path
+        return image_path
+    
+    def convert_to_answer_ids(self, responces):
+        answer_ids = []
+        for idx, responce in enumerate(responces):
+            answers = self.get_answers(idx)
+            index = answers.index(responce)
+            answer_ids.append(index)
+        return answer_ids