Spaces:

ai4data
/

data-use-extractor

Running

rafmacalaba commited on 22 days ago

Commit

6f7e99d

1 Parent(s): 2555853

Update default cached model to production ai4data/datause-extraction

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ print("Pre-downloading models to local cache...")
 try:
     from huggingface_hub import snapshot_download
     snapshot_download(repo_id="fastino/gliner2-large-v1")
-    snapshot_download(repo_id="ai4data/datause-extraction-v1")
     snapshot_download(repo_id="ai4data-use/bert-base-uncased-data-use")
     print("Models pre-downloaded successfully.")
 except Exception as e:

 try:
     from huggingface_hub import snapshot_download
     snapshot_download(repo_id="fastino/gliner2-large-v1")
+    snapshot_download(repo_id="ai4data/datause-extraction")
     snapshot_download(repo_id="ai4data-use/bert-base-uncased-data-use")
     print("Models pre-downloaded successfully.")
 except Exception as e: