Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -155,7 +155,7 @@ def predict(
|
|
| 155 |
.replace(")", "\)")
|
| 156 |
)
|
| 157 |
c = ", ".join(list(b.keys()))
|
| 158 |
-
|
| 159 |
items = rawimage.info
|
| 160 |
geninfo = ""
|
| 161 |
|
|
@@ -204,7 +204,7 @@ def predict(
|
|
| 204 |
message = "Nothing found in the image."
|
| 205 |
info = f"<div><p>{message}<p></div>"
|
| 206 |
|
| 207 |
-
return (a, c, rating, character_res, general_res, info)
|
| 208 |
|
| 209 |
|
| 210 |
def main():
|
|
@@ -248,6 +248,7 @@ def main():
|
|
| 248 |
outputs=[
|
| 249 |
gr.Textbox(label="Output (string)"),
|
| 250 |
gr.Textbox(label="Output (raw string)"),
|
|
|
|
| 251 |
gr.Label(label="Rating"),
|
| 252 |
gr.Label(label="Output (characters)"),
|
| 253 |
gr.Label(label="Output (tags)"),
|
|
|
|
| 155 |
.replace(")", "\)")
|
| 156 |
)
|
| 157 |
c = ", ".join(list(b.keys()))
|
| 158 |
+
d = c = " ".join(list(b.keys()))
|
| 159 |
items = rawimage.info
|
| 160 |
geninfo = ""
|
| 161 |
|
|
|
|
| 204 |
message = "Nothing found in the image."
|
| 205 |
info = f"<div><p>{message}<p></div>"
|
| 206 |
|
| 207 |
+
return (a, c,d, rating, character_res, general_res, info)
|
| 208 |
|
| 209 |
|
| 210 |
def main():
|
|
|
|
| 248 |
outputs=[
|
| 249 |
gr.Textbox(label="Output (string)"),
|
| 250 |
gr.Textbox(label="Output (raw string)"),
|
| 251 |
+
gr.Textbox(label="Output (booru string)"),
|
| 252 |
gr.Label(label="Rating"),
|
| 253 |
gr.Label(label="Output (characters)"),
|
| 254 |
gr.Label(label="Output (tags)"),
|