Spaces:
Running
on
Zero
Running
on
Zero
Kohaku-Blueleaf
commited on
Commit
•
36f7056
1
Parent(s):
25e6739
change some default values
Browse files
app.py
CHANGED
@@ -202,7 +202,7 @@ def numpy2pytorch(imgs):
|
|
202 |
@torch.inference_mode()
|
203 |
def interrogator_process(x):
|
204 |
img = Image.fromarray(x)
|
205 |
-
rating, features, chars = get_wd14_tags(img, general_threshold=0.
|
206 |
result = ""
|
207 |
for char in chars:
|
208 |
result += char
|
|
|
202 |
@torch.inference_mode()
|
203 |
def interrogator_process(x):
|
204 |
img = Image.fromarray(x)
|
205 |
+
rating, features, chars = get_wd14_tags(img, general_threshold=0.3, no_underline=True)
|
206 |
result = ""
|
207 |
for char in chars:
|
208 |
result += char
|