Spaces:

chuanenlin
/

foodnet

Sleeping

David Chuan-En Lin commited on Nov 29, 2021

Commit

25dd820

1 Parent(s): a618fc8

init

Files changed (1) hide show

foodnet.py CHANGED Viewed

@@ -252,7 +252,9 @@ if __name__ == "__main__":
     #   # model_path = input("Model filename and directory [eg. models/new_model.model]:   ")
     #   # model.save(model_path)
     # else:
-    model, yum = load_model('fastfood.model')
     ##### UI/UX #####

     #   # model_path = input("Model filename and directory [eg. models/new_model.model]:   ")
     #   # model.save(model_path)
     # else:
+    gdown.download('https://drive.google.com/uc?id=1fXGsWEbr-1BftKtOsnxc61cM3akMAIC0', 'fastfood.pth')
+    gdown.download('https://drive.google.com/uc?id=1h_TijdSw1K9RT3dnlfIg4xtl8WPNNQmn', 'fastfood.pth.wv.vectors_ngrams.npy')
+    model, yum = load_model('fastfood.pth')
     ##### UI/UX #####