hernanhgm commited on
Commit
86560f1
·
1 Parent(s): f8ed8e2

use joblib load instead of autotransformers

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,6 +1,10 @@
1
  import numpy as np
2
  from transformers import AutoModel
3
  import gradio as gr
 
 
 
 
4
 
5
 
6
  def promediado(L_izda, DV_izda, L_dcha, DV_dcha):
@@ -10,8 +14,8 @@ def promediado(L_izda, DV_izda, L_dcha, DV_dcha):
10
 
11
  def clasificador(L, DV):
12
 
13
- model = AutoModel.from_pretrained("hernanhgm/buitre_negro/model.pkl")
14
- scaler = AutoModel.from_pretrained("hernanhgm/buitre_negro/scaler.pkl")
15
 
16
  data = np.array([L, DV]).reshape(1, -1)
17
  data_scaled = scaler.transform(data)
 
1
  import numpy as np
2
  from transformers import AutoModel
3
  import gradio as gr
4
+ from joblib import load
5
+
6
+
7
+
8
 
9
 
10
  def promediado(L_izda, DV_izda, L_dcha, DV_dcha):
 
14
 
15
  def clasificador(L, DV):
16
 
17
+ model = load("model.pkl")
18
+ scaler = load("scaler.pkl")
19
 
20
  data = np.array([L, DV]).reshape(1, -1)
21
  data_scaled = scaler.transform(data)