sagravela commited on
Commit
813900e
·
1 Parent(s): 81f3e0e

inference update

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -5,8 +5,8 @@ import os
5
  class Inference():
6
  def __init__(self):
7
 
8
- index_path = os.path.join("recommender_model", "index")
9
- model_path = os.path.join("recommender_model", "model")
10
  self.index = tf.keras.models.load_model(index_path)
11
  self.model = tf.keras.models.load_model(model_path)
12
 
 
5
  class Inference():
6
  def __init__(self):
7
 
8
+ index_path = os.path.join("recommendation_model", "index")
9
+ model_path = os.path.join("recommendation_model", "model")
10
  self.index = tf.keras.models.load_model(index_path)
11
  self.model = tf.keras.models.load_model(model_path)
12