Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import requests
 from bokeh.plotting import figure
 from bokeh.models import HoverTool
+import torch
 import joblib
 import os
 from date_features import getDateFeatures
@@ -13,12 +14,10 @@ from date_features import getDateFeatures
 assets_dir = "assets/ML components"
 model_path = os.path.join(assets_dir, 'model.pkl')
 encoder_path = os.path.join(assets_dir, 'encoder.pkl')
-# model_path = os.path.join(current_dir, 'model.pkl')
-# encoder_path = os.path.join(current_dir, 'encoder.pkl')
 
-# Load the model and encoder from the pickle file
-model = joblib.load(model_path)
-encoder = joblib.load(encoder_path)
+# Load the model and encoder from the Hugging Face assets folder
+model = torch.hub.load(assets_dir, 'model')
+encoder = torch.hub.load(assets_dir, 'encoder')
 
 # Set Page Configurations
 st.set_page_config(page_title="ETA Prediction App", page_icon="fas fa-chart-line", layout="wide", initial_sidebar_state="auto")