orrp commited on
Commit
dafc809
·
1 Parent(s): 4c5d66a
Files changed (1) hide show
  1. vampnet/app.py +2 -4
vampnet/app.py CHANGED
@@ -1,4 +1,3 @@
1
- import os
2
  import sys
3
  import uuid
4
  from pathlib import Path
@@ -13,8 +12,7 @@ import yaml
13
  from vampnet import mask as pmask
14
  from vampnet.interface import Interface
15
 
16
- SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
17
- os.chdir(SCRIPT_DIR)
18
 
19
  # 2. Define the models directory
20
  MODEL_DIR = SCRIPT_DIR / "models"
@@ -697,4 +695,4 @@ with gr.Blocks() as demo:
697
  )
698
 
699
 
700
- demo.launch(share=True, debug=True, allowed_paths=[os.path.join(SCRIPT_DIR, "assets")])
 
 
1
  import sys
2
  import uuid
3
  from pathlib import Path
 
12
  from vampnet import mask as pmask
13
  from vampnet.interface import Interface
14
 
15
+ SCRIPT_DIR = Path()
 
16
 
17
  # 2. Define the models directory
18
  MODEL_DIR = SCRIPT_DIR / "models"
 
695
  )
696
 
697
 
698
+ demo.launch(share=True, debug=True, allowed_paths=[SCRIPT_DIR / "assets"])