nesticot commited on
Commit
a19372a
1 Parent(s): a26017d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -7,8 +7,15 @@ from datetime import datetime
7
  import numpy as np
8
  import joblib
9
 
10
- print('Starting')
11
- df_2024 = pd.read_csv('2024_spring_data.csv',index_col=[0])
 
 
 
 
 
 
 
12
  print('Starting')
13
  spring_teams = df_2024.groupby(['pitcher_id']).tail(1)[['pitcher_id','pitcher_team']].set_index(['pitcher_id'])['pitcher_team'].to_dict()
14
 
 
7
  import numpy as np
8
  import joblib
9
 
10
+ import datasets
11
+ from datasets import load_dataset
12
+ dataset = load_dataset('nesticot/mlb_data', data_files=['mlb_pitch_data_2024_spring.csv'])
13
+ dataset_train = dataset['train']
14
+ df_2024 = dataset_train.to_pandas().set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True)
15
+
16
+
17
+ # print('Starting')
18
+ # df_2024 = pd.read_csv('2024_spring_data.csv',index_col=[0])
19
  print('Starting')
20
  spring_teams = df_2024.groupby(['pitcher_id']).tail(1)[['pitcher_id','pitcher_team']].set_index(['pitcher_id'])['pitcher_team'].to_dict()
21