Hugging Face's logo

Spaces:
KBlueLeaf
/
DTG-demo
Running on Zero

App Files Community
4
KBlueLeaf commited on
Commit
d510fba
•
1 Parent(s): 293494b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ from kgen.generate import tag_gen
18
  from kgen.metainfo import SPECIAL, TARGET
19
 
20
 
21
- MODEL_PATHS = ["KBlueLeaf/DanTagGen-alpha", "KBlueLeaf/DanTagGen-beta"]
22
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
23
  print(f"Using device: {DEVICE}")
24
 
 
18
  from kgen.metainfo import SPECIAL, TARGET
19
 
20
 
21
+ MODEL_PATHS = ["KBlueLeaf/DanTagGen-alpha", "KBlueLeaf/DanTagGen-beta", "KBlueLeaf/DanTagGen-gamma"]
22
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
23
  print(f"Using device: {DEVICE}")
24