first commit

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -133,7 +133,7 @@ def get_response(image,prefix):
 
 def download_model():
     print('downloading model')
-    os.system('gdown 1-HyAeenHhS0xu2m9-fvsZw5sGhytsf7s')
+    #os.system('gdown 1-HyAeenHhS0xu2m9-fvsZw5sGhytsf7s')
 
 def show_example(path):
     return cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2RGB)
@@ -159,8 +159,8 @@ if __name__ == "__main__":
     TOKENIZER_PATH = "paligemma_tokenizer.model"
     MODEL_PATH = './traffic-vqa_ckpt_person1.npz'
 
-    # if not (os.path.exists(MODEL_PATH)):
-    #     download_model()
+    if not (os.path.exists(MODEL_PATH)):
+        download_model()
 
 
     backend = jax.lib.xla_bridge.get_backend()