David Chuan-En Lin commited on
Commit
b77f2ea
Β·
1 Parent(s): 09b3a82

Model path fix

Browse files
Files changed (1) hide show
  1. foodnet.py +2 -2
foodnet.py CHANGED
@@ -189,7 +189,7 @@ def train_model(data):
189
  return model
190
 
191
  @st.cache(allow_output_mutation=True)
192
- def load_model(filename='models/fastfood_orig_4.model'):
193
  '''
194
  Load the FastText Model
195
  :params:
@@ -254,7 +254,7 @@ st.set_page_config(page_title="FoodNet", page_icon = "πŸ”", layout = "centered"
254
  ## Sidebar ##
255
  add_selectbox = st.sidebar.selectbox("Pages", ("FoodNet Recommender", "Food Donation Resources", "Contact Team"))
256
 
257
- model, yum = load_model()
258
 
259
  if add_selectbox == "FoodNet Recommender":
260
  st.title("FoodNet πŸ”")
 
189
  return model
190
 
191
  @st.cache(allow_output_mutation=True)
192
+ def load_model(filename):
193
  '''
194
  Load the FastText Model
195
  :params:
 
254
  ## Sidebar ##
255
  add_selectbox = st.sidebar.selectbox("Pages", ("FoodNet Recommender", "Food Donation Resources", "Contact Team"))
256
 
257
+ model, yum = load_model('fastfood.pth')
258
 
259
  if add_selectbox == "FoodNet Recommender":
260
  st.title("FoodNet πŸ”")