Update app.py

app.py

@@ -1,12 +1,22 @@
-from fastai.vision.all import load_learner
-learn = load_learner('CameraTrap_model.pkl')
+import os
+import shutil
+from fastai.vision.all import *
+import gradio as gr
 
-labels = learn.dls.vocab
-def predict(img):
-    img = PILImage.create(img)
-    pred,pred_idx,probs = learn.predict(img)
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+model_path = '/content/CameraTrap_model.pkl'
+learn = load_learner(model_path)
 
+def predict_class(file_path):
+    pred, pred_idx, probs = learn.predict(file_path)
+    return str(pred)
 
-import gradio as gr
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=2)).launch(share=False)
+def classify_image(file):
+    # Move the file to the respective folders
+    if str(file) == 'With_Animals':
+        shutil.move(file, os.path.join(animal_folder, file))
+    elif str(file) == 'No_Animals':
+        shutil.move(file, os.path.join(no_animal_folder, file))
+    return 'File classified successfully!'
+
+iface = gr.Interface(fn=classify_image, inputs='file', outputs='text', title='Image Classifier')
+iface.launch()