Parthebhan commited on
Commit
eebe194
·
verified ·
1 Parent(s): e600212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import pickle
2
  import gradio as gr
3
 
4
- model = pickle.load('./Automatidata_gui.pickle')
 
 
5
 
6
  # Define the function for making predictions
7
  def automatidata(VendorID, passenger_count, Distance, Duration, rush_hour):
 
1
  import pickle
2
  import gradio as gr
3
 
4
+ # Load the pickled model
5
+ with open('./Automatidata_gui.pickle', 'rb') as file:
6
+ model = pickle.load(file)
7
 
8
  # Define the function for making predictions
9
  def automatidata(VendorID, passenger_count, Distance, Duration, rush_hour):