Update main.py

main.py

@@ -34,6 +34,11 @@ class ImageRequest(BaseModel):
 def load_assets():
     global segmentation_model, classification_model, knowledge_base, transform_segment, transform_classify
 
+    # Haddii hore loo load gareeyay, iska dhaaf
+    if segmentation_model is not None and classification_model is not None and knowledge_base is not None:
+        print("🔁 Models and knowledge base already loaded. Skipping reloading.")
+        return
+
     print("--> API starting up: This may take a few minutes...")
 
     # Use /tmp for writable cache directory
@@ -132,7 +137,7 @@ async def analyze_image(request: ImageRequest):
     confidence, predicted_idx = torch.max(probabilities, 1)
     confidence_percent = confidence.item() * 100
 
-    if confidence_percent < 75.0:
+    if confidence_percent < 50.0:
         return {"status": "Inconclusive", "message": f"Model confidence ({confidence_percent:.2f}%) is below the 75% threshold."}
 
     # Stage 3: Return result