Merge pull request #4 from SDAIA-KAUST-AI/remove_lfs

.gitattributes

@@ -1,3 +0,0 @@
-media/* filter=lfs diff=lfs merge=lfs -text
-release_ckpts/** filter=lfs diff=lfs merge=lfs -text
-demo_data/** filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -4,6 +4,8 @@
 /slurm_logs
 
 /lightning_logs/
+/demo_data/
+/release_ckpts/
 
 __pycache__
 

README.md

@@ -4,11 +4,8 @@
 
 ### Cloning the repo
 
-Install git LFS via [this instruction](https://docs.github.com/en/repositories/working-with-files/managing-large-files/installing-git-large-file-storage).
 ```bash
 git clone https://github.com/SDAIA-KAUST-AI/diabetic-retinopathy-detection.git
-git lfs install # to make sure LFS is enabled
-git lfs pull # to bring in demo images and pretrained models
 ```
 
 ### Gradio app environment
@@ -22,6 +19,8 @@ pip install -r requirements.txt
 python app.py
 ```
 
+The app will download 280 MB of files from S3 and launch.
+
 Install manually:
 
 ```bash

app.py

@@ -10,6 +10,7 @@ from transformers import (AutoImageProcessor,
                           ResNetForImageClassification)
 
 from src.labelmap import DR_LABELMAP
+from src.fetch_files import fetch_files
 
 
 class App:
@@ -22,6 +23,9 @@ class App:
 
         path = f"release_ckpts/{ckpt_name}/inference/"
 
+        if not os.path.exists(path):
+            raise Exception(f"Checkpoint not found at {path}")
+
         self.image_processor = AutoImageProcessor.from_pretrained(path)
 
         self.model = ResNetForImageClassification.from_pretrained(path)
@@ -147,6 +151,7 @@ class App:
 
 def main():
     """ App entry point. """
+    fetch_files()
     app = App()
     app.launch()