Spaces:

westlake-repl
/

Demo_ProTrek_650M_UniRef50

Running

LTEnjoy commited on Aug 9, 2024

Commit

8fdbefd

1 Parent(s): 1c4a4a8

Update

Files changed (3) hide show

Dockerfile CHANGED Viewed

@@ -24,10 +24,10 @@ RUN wget 'https://drive.google.com/file/d/1waxLyL2MLCxyHyIlQe7zkeHy8lzuDThu/view
 RUN chmod +x /tmp/TMalign
 # Download ProTrek model
-RUN huggingface-cli download westlake-repl/ProTrek_650M_UniRef50 --repo-type model --local-dir /tmp/ProTrek_650M_UniRef50
 # Download ProTrek faiss index
-RUN huggingface-cli download westlake-repl/ProTrek-faiss-index  --repo-type dataset  --local-dir /tmp/ProTrek-faiss-index --include "ProTrek_650M_UniRef50/Swiss-Prot/*"
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user

 RUN chmod +x /tmp/TMalign
 # Download ProTrek model
+RUN huggingface-cli download westlake-repl/ProTrek_650M_UniRef50 --repo-type model --local-dir /data/ProTrek_650M_UniRef50
 # Download ProTrek faiss index
+RUN huggingface-cli download westlake-repl/ProTrek-faiss-index  --repo-type dataset  --local-dir /data/ProTrek-faiss-index --include "ProTrek_650M_UniRef50/Swiss-Prot/*"
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user

demo/config.yaml CHANGED Viewed

@@ -1,14 +1,14 @@
-model_dir: /tmp/ProTrek_650M_UniRef50
 faiss_config:
   IO_FLAG_MMAP: True
 sequence_index_dir:
   - name: Swiss-Prot
-    index_dir: /tmp/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/sequence
 structure_index_dir:
   - name: Swiss-Prot
-    index_dir: /tmp/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/structure
 text_index_dir:
   - name: Swiss-Prot
-    index_dir: /tmp/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/text

+model_dir: /data/ProTrek_650M_UniRef50
 faiss_config:
   IO_FLAG_MMAP: True
 sequence_index_dir:
   - name: Swiss-Prot
+    index_dir: /data/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/sequence
 structure_index_dir:
   - name: Swiss-Prot
+    index_dir: /data/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/structure
 text_index_dir:
   - name: Swiss-Prot
+    index_dir: /data/ProTrek-faiss-index/ProTrek_650M_UniRef50/Swiss-Prot/text

run.py CHANGED Viewed

@@ -37,7 +37,7 @@ async def run_cmd(cmd):
 async def download_faiss_index():
     await asyncio.gather(
         run_cmd("echo Hello World!"),
-        run_cmd('huggingface-cli download westlake-repl/ProTrek-faiss-index  --repo-type dataset  --local-dir /tmp/ProTrek-faiss-index_UniRef50 --include "ProTrek_650M_UniRef50/UniRef50/*"'),
     )

 async def download_faiss_index():
     await asyncio.gather(
         run_cmd("echo Hello World!"),
+        run_cmd('huggingface-cli download westlake-repl/ProTrek-faiss-index  --repo-type dataset  --local-dir /data/ProTrek-faiss-index --include "ProTrek_650M_UniRef50/UniRef50/*"'),
     )