Spaces:
Sleeping
Sleeping
File size: 533 Bytes
220b4dd 205d5cc 220b4dd 6ce82f5 ffec641 220b4dd 205d5cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from prediction import ModelPrediction
import utilities as us
def main():
model = ModelPrediction()
response = model.make_prediction(question='What is the name of Simone', db_schema='CREATE TABLE Player(Name, Age)', model_name="gpt-3.5", prompt='{question} {db_schema}')
print(response) # dict[response, response_parsed, cost]
if __name__ == "__main__":
# main()
us.check_and_create_dir('data/data_interface/')
us.check_and_create_dir('data/data_results/')
us.check_and_create_dir('data/databases/')
|