Spaces:
Sleeping
Sleeping
File size: 463 Bytes
220b4dd 205d5cc 220b4dd 2792e5a ffec641 220b4dd 2792e5a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from prediction import ModelPrediction
import utilities as us
def main():
model = ModelPrediction()
response = model.make_prediction(
question='What is the surnamname of Simone',
db_schema='CREATE TABLE Player(Name, Age)\nINSERT INTO Player(Name, Age) VALUES (Simone, Papicchio)',
model_name="gpt-3.5",
task='QA'
)
print(response) # dict[response, response_parsed, cost]
if __name__ == "__main__":
main()
|