diff --git a/__pycache__/main_classify_fish.cpython-37.pyc b/__pycache__/main_classify_fish.cpython-37.pyc
new file mode 100644
index 0000000..a7bfafb
Binary files /dev/null and b/__pycache__/main_classify_fish.cpython-37.pyc differ
diff --git a/__pycache__/main_detect_fish.cpython-37.pyc b/__pycache__/main_detect_fish.cpython-37.pyc
new file mode 100644
index 0000000..ac044b5
Binary files /dev/null and b/__pycache__/main_detect_fish.cpython-37.pyc differ
diff --git a/main.py b/main.py
index 82a69a8..e679bd6 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,8 @@
 from flask import Flask, render_template, request, redirect, flash, url_for, render_template_string
 import urllib.request
 from werkzeug.utils import secure_filename
-from fishdetectionui.main_detect_fish import obj_model_predict
-from fishdetectionui.main_classify_fish import getPrediction
+from main_detect_fish import obj_model_predict
+from main_classify_fish import getPrediction
 import os
 import numpy as np
 import cv2