fix error in prediction.py

prediction.py

@@ -23,7 +23,7 @@ import litellm
 
 
 class ModelPrediction:
-    def __init__(self, model_name):
+    def __init__(self):
         self.model_name2pred_func = {
             "gpt-3.5": self._model_prediction("gpt-3.5"),
             "gpt-4o-mini": self._model_prediction("gpt-4o-mini"),

test_prediction.py

@@ -0,0 +1,10 @@
+from prediction import ModelPrediction
+
+
+def main():
+    model = ModelPrediction()
+    response = model.make_prediction("Hi, how are you?", "gpt-3.5")
+    print(response)
+
+if __name__ == "__main__":
+    main()