diff --git a/.gitignore b/.gitignore
index d6b1547..29784c9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,5 @@
 .vscode/
-*/__pycache__/
+*__pycache__*
 *tex
 Images/TrainingImages/
 Model/bottleneck/
diff --git a/Logic/TrashRecognition/ImageClassification.py b/Logic/TrashRecognition/ImageClassification.py
index 72b29d3..ebd1686 100644
--- a/Logic/TrashRecognition/ImageClassification.py
+++ b/Logic/TrashRecognition/ImageClassification.py
@@ -199,7 +199,7 @@ if __name__ == "__main__":
   if args.output_layer:
     output_layer = args.output_layer
   
-  classify(model_file=model_file, label_file=label_file, input_height=input_height, input_width=input_width,
+  classify_files(model_file=model_file, label_file=label_file, input_height=input_height, input_width=input_width,
     input_mean=input_mean, input_std=input_std, input_layer=input_layer, output_layer=output_layer)
   
     # for i in top_k: