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()