new world

app.py

@@ -3,14 +3,14 @@ import onnxruntime as rt
 from transformers import AutoTokenizer
 import torch, json
 
-tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
+tokenizer = AutoTokenizer.from_pretrained("roberta-base")
 
-with open("recipe-3_types_encoded.json", "r") as fp:
+with open("recipe_types_encoded19.json", "r") as fp:
   encode_types = json.load(fp)
 
 recipe = list(encode_types.keys())
 
-inf_session = rt.InferenceSession('recipe3-classifier.onnx')
+inf_session = rt.InferenceSession('recipe-classifier.onnx')
 input_name = inf_session.get_inputs()[0].name
 output_name = inf_session.get_outputs()[0].name
 

recipe-3_types_encoded.json

@@ -1 +0,0 @@
-{"low-carb": 0, "Cheese": 1, "Vegetable": 2, "Low-cholesterol": 3, "Gluten-free": 4, "nutritional": 5, "High-protein": 6, "Baking": 7, "pastry": 8, "high-fiber": 9, "Vegan": 10, "diet": 11, "Pasta": 12, "Fish": 13, "High-fiber Vegetarian Dish": 14, "Beef": 15, "salad": 16, "Eggs": 17, "Vegetarian": 18, "saltwater fish": 19, "fruit-vegetable": 20, "Vegetable Salad": 21, "flowering vegetables": 22, "Meat": 23, "Sauce": 24, "Asian": 25, "Chicken": 26, "Dessert": 27, "Herb": 28, "baked dish": 29, "Roast": 30, "Legume": 31, "Shrimp": 32, "Italian": 33, "soup": 34, "Seafood": 35, "non-alcoholic": 36, "Casserole": 37, "mushroom": 38, "Spices": 39, "leafy green vegetable": 40, "Sprout": 41, "Fruit": 42, "Tropical Fruit": 43, "Dough": 44, "Rice": 45, "Pome Fruit": 46, "Fruit Dessert": 47}

recipe3-classifier.onnx → recipe-classifier-stage-18.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8422ac641b65e11a5de7aa537fb7201b794d5606f49c3fecf26d6ca7cf4dbe2b
-size 328773212
+oid sha256:ee07d4871090baf7cbfc1096cbd15a3b64b54164bf66ced8769f660c17c1396d
+size 500811720

recipe-classifier.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84e305096a173db27b4b760d7b996451c1704f037c56f07c2bc6d850c2e5f338
+size 499042987

recipe_types_encoded19.json

@@ -0,0 +1 @@
+{"Vegetable": 0, "Cheese": 1, "low-carb": 2, "non-alcoholic": 3, "egg-free": 4, "Low-cholesterol": 5, "250-400 Calorie": 6, "Gluten-free": 7, "nutritional": 8, "High-protein": 9, "Mineral-rich": 10, "lactose-free": 11, "100-250 Calorie": 12, "pastry": 13, "North America": 14, "Baking": 15, "high-fiber": 16, "Vegan": 17, "diet": 18, "Pasta": 19, "meat-free": 20, "Fruit": 21, "Fish": 22, "Vegetarian": 23, "Beef": 24, "mushroom": 25, "Meat": 26, "salad": 27, "Eggs": 28, "Seafood": 29, "soup": 30, "German": 31, "Dessert": 32, "Low-sugar": 33, "saltwater fish": 34, "fruit-vegetable": 35, "Vegetable Salad": 36, "400-500 Calorie": 37, "Chicken": 38, "Asian": 39, "Tropical Fruit": 40, "Appetizer": 41, "Roast": 42, "Shrimp": 43, "Sprout": 44, "Shellfish": 45, "Low-fat": 46, "Italian": 47, "Pork": 48, "American": 49, "Dairy": 50, "Dough": 51, "Rice": 52, "Pome Fruit": 53, "Fruit Dessert": 54}