fixing prediction

model.py

@@ -225,7 +225,7 @@ Category:"""
                             "choices": [best_choice]
                         }
                     }],
-                    "model_version": self.get("model_version", "1.0.0")
+                    "model_version": "1.0.0"
                 }
                 logger.info(f"Formatted prediction: {json.dumps(prediction, indent=2)}")
                 predictions.append(prediction)
@@ -235,7 +235,7 @@ Category:"""
             raise
         
         logger.info(f"Returning {len(predictions)} predictions")
-        return predictions  # Changed from ModelResponse to match Label Studio's expectation
+        return predictions
     
     def fit(self, event, data, **kwargs):
         """Fine-tune the model using LoRA when new annotations are available"""