Update app.py

app.py

@@ -11,11 +11,20 @@ def predict_class(file_path):
     return str(pred)
 
 def classify_image(file):
+    # Create your own destination folders
+    animal_folder = 'path/to/animal_folder'
+    no_animal_folder = 'path/to/no_animal_folder'
+
+    os.makedirs(animal_folder, exist_ok=True)
+    os.makedirs(no_animal_folder, exist_ok=True)
+
     # 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))
+    pred_class = predict_class(file.name)
+    if pred_class == 'With_Animals':
+        shutil.move(file.name, os.path.join(animal_folder, file.name))
+    elif pred_class == 'No_Animals':
+        shutil.move(file.name, os.path.join(no_animal_folder, file.name))
+    
     return 'File classified successfully!'
 
 iface = gr.Interface(fn=classify_image, inputs='file', outputs='text', title='Image Classifier From CameraTrap')