ikoghoemmanuell
commited on
Commit
•
03c817f
1
Parent(s):
82f42f7
Update app.py
Browse files
app.py
CHANGED
@@ -6,17 +6,12 @@ import requests
|
|
6 |
from bokeh.plotting import figure
|
7 |
from bokeh.models import HoverTool
|
8 |
import pickle
|
9 |
-
import torch
|
10 |
from date_features import getDateFeatures
|
11 |
|
12 |
# Define path for the model and encoder from the pickle file
|
13 |
model = pickle.load(open("model.pkl", "rb"))
|
14 |
encoder = pickle.load(open("encoder.pkl", "rb"))
|
15 |
|
16 |
-
# Load the model and encoder from the Hugging Face assets folder
|
17 |
-
model = torch.hub.load(assets_dir, 'model')
|
18 |
-
encoder = torch.hub.load(assets_dir, 'encoder')
|
19 |
-
|
20 |
# Set Page Configurations
|
21 |
st.set_page_config(page_title="ETA Prediction App", page_icon="fas fa-chart-line", layout="wide", initial_sidebar_state="auto")
|
22 |
|
|
|
6 |
from bokeh.plotting import figure
|
7 |
from bokeh.models import HoverTool
|
8 |
import pickle
|
|
|
9 |
from date_features import getDateFeatures
|
10 |
|
11 |
# Define path for the model and encoder from the pickle file
|
12 |
model = pickle.load(open("model.pkl", "rb"))
|
13 |
encoder = pickle.load(open("encoder.pkl", "rb"))
|
14 |
|
|
|
|
|
|
|
|
|
15 |
# Set Page Configurations
|
16 |
st.set_page_config(page_title="ETA Prediction App", page_icon="fas fa-chart-line", layout="wide", initial_sidebar_state="auto")
|
17 |
|