protectai / modelscan

Protection against Model Serialization Attacks

Home Page:http://modelscan.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add is_compatible API to ModelScan SDK

willarmiros opened this issue · comments

Today, ModelScan checks file extensions to determine if a file is able to be scanned. It would be nice to have an API that just does this check, so that we don't have to perform an entire scan to see if a file is compatible or not. Add a is_compatible(path: str) -> bool API to ModelScan that just checks the extension for compatibility.