Update app.py

app.py

@@ -12,6 +12,9 @@ def load_model():
     model.to(device)
     return model
 
+# Load the model outside caching
+model = load_model()  # Place this line here, right after defining load_model
+
 @st.cache_data
 def read_image_as_np_array(image_path):
     if "http" in image_path:
@@ -20,8 +23,6 @@ def read_image_as_np_array(image_path):
         image = Image.open(image_path).convert("L").convert("RGB")
     image = np.array(image)
     return image
-    
-model = load_model()    
 
 @st.cache_data
 def predict_detections_and_associations(
@@ -44,6 +45,7 @@ def predict_detections_and_associations(
             )[0]
     return result
 
+
 @st.cache_data
 def predict_ocr(
     image_path,