OrlandoHugBot commited on
Commit
dd441ec
·
verified ·
1 Parent(s): 9ac11e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,8 +33,8 @@ except ImportError:
33
  sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
34
 
35
  # Model configuration
36
- MODEL_NAME = os.environ.get("MODEL_NAME", "/data_genie/genie/chris/Unipic3-DMD")
37
- TRANSFORMER_PATH = os.environ.get("TRANSFORMER_PATH", "/data_genie/genie/chris/Unipic3-DMD/ema_transformer")
38
 
39
  dtype = torch.bfloat16
40
 
 
33
  sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
34
 
35
  # Model configuration
36
+ MODEL_NAME = os.environ.get("MODEL_NAME", "Skywork/Unipic3-DMD")
37
+ TRANSFORMER_PATH = os.environ.get("TRANSFORMER_PATH", "Skywork/Unipic3-DMD/ema_transformer")
38
 
39
  dtype = torch.bfloat16
40