Update src/submission/check_validity.py

src/submission/check_validity.py

@@ -56,7 +56,10 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
         )
 
     except Exception as e:
-        return False, f"was not found on hub!({e})", None
+        traceback.print_exc()
+        if "You are trying to access a gated repo." in str(e):
+            return True, "uses a gated model.", None
+        return False, "was not found on hub!", None
 
 
 def get_model_size(model_info: ModelInfo, precision: str):