NeerAbhy commited on
Commit
e4ef1d0
·
verified ·
1 Parent(s): 383874a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from huggingface_hub import InferenceClient
3
  import numpy as np
4
  import pandas as pd
 
5
 
6
 
7
  import pickle
@@ -12,9 +13,9 @@ For more information on `huggingface_hub` Inference API support, please check th
12
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
13
  cols = pd.read_csv('./Columns.csv')
14
  cols = cols['Cols']
15
- with open("./model.pkl", "rb") as f:
16
- model = pickle.load(f)
17
-
18
 
19
  severityDictionary=dict()
20
  description_list = dict()
 
2
  from huggingface_hub import InferenceClient
3
  import numpy as np
4
  import pandas as pd
5
+ import joblib
6
 
7
 
8
  import pickle
 
13
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
14
  cols = pd.read_csv('./Columns.csv')
15
  cols = cols['Cols']
16
+ #with open("./model.pkl", "rb") as f:
17
+ # model = pickle.load(f)
18
+ model = joblib.load("./model.pkl")
19
 
20
  severityDictionary=dict()
21
  description_list = dict()