fix

app.py

@@ -4,6 +4,10 @@ import gradio as gr
 import numpy as np
 from fastai.vision.all import *
 
+
+def label(file_name):
+    return train_labels[file_name.replace(".jpg", "")]
+
 config = {
     "labels": [
         "Plantation",
@@ -13,11 +17,12 @@ config = {
     "size": 256,
 }
 
+
+
 learn = load_learner("model.pkl")
 
 def classify_image(input):
     _, _, prediction = learn.predict(input)
-    print(prediction)
     outputs = {label: float(prediction[i]) for i, label in enumerate(config["labels"])}
     # Get argmax
     argmax_label = config["labels"][np.argmax(prediction)]
@@ -31,7 +36,7 @@ gr.Interface(
         gr.outputs.Textbox(label="Output of the model"),
         gr.outputs.Textbox(label="Probability (0 - 100)")
     ],
-    examples=[str(x) for x in Path("./").glob("*.jpg")],
+    examples=[str(x) for x in Path("./").glob("*.png")],
     flagging_options=["Correct label", "Incorrect label"],
     allow_flagging="manual",
 ).launch()