test_prediction.py

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