Fix imports, model loading and dependencies

app.py

@@ -1,15 +1,32 @@
+import sys
+import os
 import gradio as gr
 import base64
 import io
 from PIL import Image
-from medimageinsight import MedImageInsight
+from huggingface_hub import snapshot_download
 
+# Download the full lion-ai/MedImageInsights repo (code + model weights)
+# Cached after first run — won't re-download on restarts
+print("Downloading MedImageInsights repo...")
+repo_path = snapshot_download("lion-ai/MedImageInsights")
+print(f"Downloaded to: {repo_path}")
+
+# Make the repo's Python modules importable
+sys.path.insert(0, repo_path)
+
+from medimageinsightmodel import MedImageInsight  # noqa: E402
+
+model_dir = os.path.join(repo_path, "2024.09.27")
+
+print("Loading model...")
 classifier = MedImageInsight(
-    model_dir="2024.09.27",
+    model_dir=model_dir,
     vision_model_name="medimageinsigt-v1.0.0.pt",
     language_model_name="language_model.pth"
 )
 classifier.load_model()
+print("Model ready.")
 
 
 def analyze(image: Image.Image, labels_str: str):

requirements.txt

@@ -1,4 +1,11 @@
-git+https://huggingface.co/lion-ai/MedImageInsights
 huggingface_hub
 Pillow
 torch
+pandas
+timm
+transformers
+einops
+ftfy
+sentencepiece
+fvcore
+mup