github/autoMate
1
0
Fork 0
mirror of https://github.com/yuruotong1/autoMate.git synced 2025-12-26 05:16:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

delete flo

Browse Source
This commit is contained in:
yuruo 2025-03-19 22:06:52 +08:00
parent a48ff2d37d
commit d8d4a1fb5b
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gradio_ui/app.py
View File

@ -14,7 +14,7 @@ from gradio_ui.loop import (
import base64
from xbrain.utils.config import Config
from util.download_weights import OMNI_PARSER_DIR, FLORENCE_DIR
from util.download_weights import OMNI_PARSER_DIR
CONFIG_DIR = Path("~/.anthropic").expanduser()
API_KEY_FILE = CONFIG_DIR / "api_key"

5
util/download_weights.py
View File

@ -3,7 +3,6 @@ from pathlib import Path
__WEIGHTS_DIR = Path("weights")
OMNI_PARSER_DIR = os.path.join(__WEIGHTS_DIR, "AI-ModelScope", "OmniParser-v2___0")
FLORENCE_DIR = os.path.join(__WEIGHTS_DIR, "AI-ModelScope", "Florence-2-base-ft")
def download():
from modelscope import snapshot_download
# Create weights directory
@ -13,9 +12,5 @@ def download():
cache_dir='weights',
)
snapshot_download(
'AI-ModelScope/Florence-2-base-ft',
cache_dir='weights' )
if __name__ == "__main__":
download()