Spaces:

A-Celsius
/

ADR_Predictor

Runtime error

A-Celsius commited on Aug 22, 2023

Commit

cfb14a1

1 Parent(s): 72d2862

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
-import pickle
 import gradio as gr
 from datetime import datetime, timedelta, timezone
-model = pickle.load(open('model.pkl','rb'))
 def preprocess_city(selected_city):
     # Map the selected city to its one-hot encoded representation

+import pickle, joblib
 import gradio as gr
 from datetime import datetime, timedelta, timezone
+model = joblib.load('model.pkl')
 def preprocess_city(selected_city):
     # Map the selected city to its one-hot encoded representation