Spaces:
Running
Running
David Chuan-En Lin
commited on
Commit
Β·
b77f2ea
1
Parent(s):
09b3a82
Model path fix
Browse files- 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
|
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 π")
|