aopp

app.py

@@ -11,10 +11,13 @@ dls_path = "dls_origin-classifier_v1.pkl"
 
 learner_inf = load_learner(model_path)
 
+
 dls = torch.load(dls_path)
 
+
 class_label_mapping = {label: idx for idx, label in enumerate(learner_inf.dls.vocab)}
 
+
 def predict_text(text):
     prediction = learner_inf.blurr_predict(text)
     predicted_probs = prediction[0]['scores']
@@ -24,10 +27,11 @@ def predict_text(text):
 
 iface = gr.Interface(
     fn=predict_text,
-    inputs=gr.inputs.Textbox(lines=2, placeholder='Enter Recipe Here...'),
-    outputs=gr.outputs.Label(num_top_classes=3),
+    inputs="text",
+    outputs="text",
     title="Food Origin Classification App",
     description="Enter a Recipe, and it will predict the class label.",
 )
 
+
 iface.launch()