Commit
•
f56e0f7
1
Parent(s):
7f20010
Update app.py
Browse filesONNX model file in the model repo has been renamed
app.py
CHANGED
|
@@ -32,7 +32,7 @@ Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
|
|
| 32 |
|
| 33 |
HF_TOKEN = os.environ["HF_TOKEN"]
|
| 34 |
MODEL_REPO = "SmilingWolf/wd-v1-4-vit-tagger"
|
| 35 |
-
MODEL_FILENAME = "
|
| 36 |
LABEL_FILENAME = "selected_tags.csv"
|
| 37 |
|
| 38 |
|
|
|
|
| 32 |
|
| 33 |
HF_TOKEN = os.environ["HF_TOKEN"]
|
| 34 |
MODEL_REPO = "SmilingWolf/wd-v1-4-vit-tagger"
|
| 35 |
+
MODEL_FILENAME = "model.onnx"
|
| 36 |
LABEL_FILENAME = "selected_tags.csv"
|
| 37 |
|
| 38 |
|