Share vit checkpoint with RAG assistant project

models/vit-finetuned/model.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b5249456310ed9b3884807fc9a172980c3950ec008aa89dadea0d97e5d9da5e4
-size 343334720

src/app_gradio.py

@@ -6,6 +6,7 @@ import json
 import base64
 from transformers import AutoImageProcessor, AutoModelForImageClassification
 import sys
+from model_paths import resolve_model_dir
 
 # Debug flag: set to True to enable, False to disable
 DEBUG = False
@@ -16,7 +17,7 @@ def dbg(msg: str):
         print(f"[DEBUG] {msg}")
 
 
-MODEL_DIR = "models/vit-finetuned"
+MODEL_DIR = resolve_model_dir()
 
 # Build a safe path to the logo and expose it via Gradio's file= URL
 LOGO_REL = "images/plant-disease-logo.png"

src/model_paths.py

@@ -0,0 +1,36 @@
+import os
+
+
+def resolve_model_dir() -> str:
+    """
+    Determine which vit-finetuned folder to use.
+    Prefers the shared copy in HF-Plant-Disease-RAG-Assistant if present,
+    otherwise falls back to this repository's models directory.
+    """
+    override = os.getenv("MODEL_DIR")
+    if override:
+        return os.path.abspath(override)
+
+    repo_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+
+    shared_dir = os.path.abspath(
+        os.path.join(
+            repo_root,
+            "..",
+            "HF-Plant-Disease-RAG-Assistant",
+            "models",
+            "vit-finetuned",
+        )
+    )
+    shared_model = os.path.join(shared_dir, "model.safetensors")
+    if os.path.exists(shared_model):
+        return shared_dir
+
+    local_dir = os.path.join(repo_root, "models", "vit-finetuned")
+    local_model = os.path.join(local_dir, "model.safetensors")
+    if os.path.exists(local_model):
+        return local_dir
+
+    raise FileNotFoundError(
+        "model.safetensors not found in shared or local model directories."
+    )

src/streamlit_app.py

@@ -14,9 +14,10 @@ else:
     import os
     import json
     from transformers import AutoImageProcessor, AutoModelForImageClassification
+    from model_paths import resolve_model_dir
 
     # Model directory
-    MODEL_DIR = "models/vit-finetuned"
+    MODEL_DIR = resolve_model_dir()
 
     st.title("Plant Disease Classifier")
     st.write("App is starting...")