fix: numba caching

main.py

@@ -1,13 +1,14 @@
-import sys
 import os
-os.environ["NUMBA_DISABLE_JIT"] = "1"
+# os.environ["NUMBA_DISABLE_JIT"] = "1"
+os.environ["NUMBA_CACHE_DIR"] = "/tmp/numba_cache"
+
+import sys
 
 # Add the parent directory to Python path for imports
 current_dir = os.path.dirname(os.path.abspath(__file__))
 parent_dir = os.path.dirname(current_dir)
 if parent_dir not in sys.path:
     sys.path.insert(0, parent_dir)
-
 from fastapi import FastAPI, File, UploadFile, HTTPException
 import uvicorn
 import numpy as np
@@ -20,16 +21,16 @@ from PIL import Image
 from fastapi.middleware.cors import CORSMiddleware
 
 # Try different import methods for download_model
-try:
-    from API.download_model import download_model
-except ImportError:
-    try:
-        from download_model import download_model
-    except ImportError:
-        # If all else fails, define a simple fallback
-        def download_model():
-            print("Model download function not available")
-            pass
+# try:
+#     from API.download_model import download_model
+# except ImportError:
+#     try:
+#         from download_model import download_model
+#     except ImportError:
+#         # If all else fails, define a simple fallback
+#         def download_model():
+#             print("Model download function not available")
+#             pass
 
 # Initialize FastAPI app
 app = FastAPI(
@@ -48,10 +49,11 @@ app.add_middleware(
 )
 
 # Download model if not exists
-download_model()
+# download_model()
 
 # Loading the trained model
 MODEL = tf.keras.models.load_model("./model/model_inceptionV3.h5")
+print("Loaded model:", type(MODEL))
 
 # Class names (High activity or Low activity)
 CLASS_NAMES = ["High activity", "Low activity"]