Update app.py
Browse files
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 |
-
|
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()
|