all files

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from transformers import AutoTokenizer
+import torch
+from fastai.text.all import *
+
+
+# Define the path to your model
+model_path = "models/origin-classifier-stage-2.pkl"
+
+# Load the learner
+learner_inf = load_learner(model_path)
+
+# Create a mapping from class labels to indices
+class_label_mapping = {label: idx for idx, label in enumerate(learner_inf.dls.vocab)}
+
+# Define a function to make predictions
+def predict_text(text):
+    prediction = learner_inf.blurr_predict(text)[0]
+    predicted_class_index = prediction['class_index']
+    predicted_class_label = list(class_label_mapping.keys())[list(class_label_mapping.values()).index(predicted_class_index)]
+    return predicted_class_label
+
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=predict_text,
+    inputs="text",
+    outputs="text",
+    title="FoodOrigin Classification App",
+    description="Enter a text, and it will predict the class label.",
+)
+
+# Start the Gradio app
+iface.launch()

label_types_encoded.json

@@ -0,0 +1 @@
+{"Amish and Mennonite": 0, "Argentinian": 1, "Australian and New Zealander": 2, "Austrian": 3, "Bangladeshi": 4, "Belgian": 5, "Brazilian": 6, "Cajun and Creole": 7, "Canadian": 8, "Chilean": 9, "Chinese": 10, "Colombian": 11, "Cuban": 12, "Danish": 13, "Dutch": 14, "Filipino": 15, "Finnish": 16, "French": 17, "German": 18, "Greek": 19, "Indian": 20, "Indonesian": 21, "Israeli": 22, "Italian": 23, "Jamaican": 24, "Japanese": 25, "Jewish": 26, "Korean": 27, "Lebanese": 28, "Malaysian": 29, "Norwegian": 30, "Pakistani": 31, "Persian": 32, "Peruvian": 33, "Polish": 34, "Portuguese": 35, "Puerto Rican": 36, "Russian": 37, "Scandinavian": 38, "Soul Food": 39, "South African": 40, "Southern Recipes": 41, "Spanish": 42, "Swedish": 43, "Swiss": 44, "Tex-Mex": 45, "Thai": 46, "Turkish": 47, "Vietnamese": 48, "Australian and New Zealander Recipes": 49, "Cajun and Creole Recipes": 50, "Canadian Recipes": 51, "Chinese Recipes": 52, "Filipino Recipes": 53, "French Recipes": 54, "German Recipes": 55, "Greek Recipes": 56, "Indian Recipes": 57, "Italian Recipes": 58, "Japanese Recipes": 59, "Jewish Recipes": 60, "Korean Recipes": 61, "Polish Recipes": 62, "Scandinavian Recipes": 63, "Spanish Recipes": 64, "Thai Recipes": 65, "Vietnamese Recipes": 66}

origin-classifier-stage-2.pkl

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