File size: 1,036 Bytes
2532605 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | import requests
import json
print('=' * 70)
print('TEST: Natural Language Prediction (Multiple Formats)')
print('=' * 70)
test_queries = [
'Verstappen Bahrain 2023 win probability?',
'Max Verstappen Bahrain Grand Prix 2023',
'Round 1 2023 Verstappen',
'VER Bahrain 2023',
]
for query in test_queries:
print('\nTrying: ' + query)
try:
resp = requests.post(
'http://127.0.0.1:8000/predict/f1',
json={'query': query},
timeout=30
)
result = resp.json()
if 'win_probability' in result:
print(' ✓ SUCCESS!')
print(' Win probability: {}'.format(result.get('win_probability')))
print(' Driver: {} ({})'.format(
result.get('metadata', {}).get('driver_name'),
result.get('metadata', {}).get('driver_id')
))
else:
print(' ✗ Error: {}'.format(result.get('detail')))
except Exception as e:
print(' ✗ Exception: {}'.format(str(e)))
|