-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
92 lines (76 loc) · 1.92 KB
/
app.py
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import streamlit as st
import pickle
import pandas as pd
import numpy as np
pipe = pickle.load(open('pipe.pkl', 'rb'))
teams = [
'Australia',
'India',
'Bangladesh',
'New Zealand',
'South Africa',
'England',
'West Indies',
'Afghanistan',
'Pakistan',
'Sri Lanka'
]
cities = ['Colombo',
'Mirpur',
'Johannesburg',
'Dubai',
'Auckland',
'Cape Town',
'London',
'Pallekele',
'Barbados',
'Sydney',
'Melbourne',
'Durban',
'St Lucia',
'Wellington',
'Lauderhill',
'Hamilton',
'Centurion',
'Manchester',
'Abu Dhabi',
'Mumbai',
'Nottingham',
'Southampton',
'Mount Maunganui',
'Chittagong',
'Kolkata',
'Lahore',
'Delhi',
'Nagpur',
'Chandigarh',
'Adelaide',
'Bangalore',
'St Kitts',
'Cardiff',
'Christchurch',
'Trinidad']
st.title('Cricket Score Predictor')
col1, col2 = st.columns(2)
with col1:
batting_team = st.selectbox('Select batting team', sorted(teams))
with col2:
bowling_team = st.selectbox('Select bowling team', sorted(teams))
city = st.selectbox('Select city', sorted(cities))
col3,col4,col5 = st.columns(3)
with col3:
current_score = st.number_input('Current Score')
with col4:
overs = st.number_input('Overs Done (works for over > 5)')
with col5:
wickets = st.number_input('Wickets Out')
last_five = st.number_input("Runs scored in last 5 overs")
if st.button('Predict Score'):
balls_left = 120 - (overs * 6)
wickets_left = 10 - wickets
crr = current_score/overs
input_df = pd.DataFrame(
{'batting_team': [batting_team], 'bowling_team': [bowling_team], 'city': city, 'current_score': [current_score],
'balls_left': [balls_left], 'wickets_left': [wickets], 'crr': [crr], 'last_five': [last_five]})
result = pipe.predict(input_df)
st.header("Predicted Score - " + str(int(result[0])))