all files

app.py

@@ -4,12 +4,16 @@ import torch
 from fastai.text.all import *
 
 
-# Define the path to your model
-model_path = "models/origin-classifier-stage-2.pkl"
+# Define the path to your model and dataloaders
+model_path = "origin-classifier-stage-2.pkl"
+dls_path = "dls_origin-classifier_v1.pkl"
 
 # Load the learner
 learner_inf = load_learner(model_path)
 
+# Load the DataLoaders
+dls = torch.load(dls_path)
+
 # Create a mapping from class labels to indices
 class_label_mapping = {label: idx for idx, label in enumerate(learner_inf.dls.vocab)}
 
@@ -25,9 +29,10 @@ iface = gr.Interface(
     fn=predict_text,
     inputs="text",
     outputs="text",
-    title="FoodOrigin Classification App",
+    title="Text Classification App",
     description="Enter a text, and it will predict the class label.",
 )
 
 # Start the Gradio app
 iface.launch()
+

dls_origin-classifier_v1.pkl

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