Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -95,6 +95,7 @@ def predict(image: PIL.Image.Image, score_threshold: float,
|
|
95 |
res[label] = prob
|
96 |
b = dict(sorted(res.items(),key=lambda item:item[1], reverse=True))
|
97 |
a = ', '.join(list(b.keys())).replace('_',' ').replace('(','\(').replace(')','\)')
|
|
|
98 |
|
99 |
items = rawimage.info
|
100 |
geninfo = ''
|
@@ -131,7 +132,7 @@ def predict(image: PIL.Image.Image, score_threshold: float,
|
|
131 |
message = "Nothing found in the image."
|
132 |
info = f"<div><p>{message}<p></div>"
|
133 |
|
134 |
-
return (
|
135 |
|
136 |
|
137 |
def main():
|
@@ -153,9 +154,10 @@ def main():
|
|
153 |
label='Score Threshold'),
|
154 |
],
|
155 |
[
|
156 |
-
gr.outputs.
|
157 |
-
gr.outputs.Textbox(label='
|
158 |
-
gr.outputs.Label(label='
|
|
|
159 |
],
|
160 |
examples=[
|
161 |
['miku.jpg',0.5],
|
@@ -166,6 +168,7 @@ def main():
|
|
166 |
Demo for [KichangKim/DeepDanbooru](https://github.com/KichangKim/DeepDanbooru) with "ready to copy" prompt and a prompt analyzer.
|
167 |
|
168 |
Modified from [hysts/DeepDanbooru](https://huggingface.co/spaces/hysts/DeepDanbooru)
|
|
|
169 |
PNG Info code forked from [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
170 |
''',
|
171 |
theme=args.theme,
|
|
|
95 |
res[label] = prob
|
96 |
b = dict(sorted(res.items(),key=lambda item:item[1], reverse=True))
|
97 |
a = ', '.join(list(b.keys())).replace('_',' ').replace('(','\(').replace(')','\)')
|
98 |
+
c = ', '.join(list(b.keys()))
|
99 |
|
100 |
items = rawimage.info
|
101 |
geninfo = ''
|
|
|
132 |
message = "Nothing found in the image."
|
133 |
info = f"<div><p>{message}<p></div>"
|
134 |
|
135 |
+
return (a,c,res,info)
|
136 |
|
137 |
|
138 |
def main():
|
|
|
154 |
label='Score Threshold'),
|
155 |
],
|
156 |
[
|
157 |
+
gr.outputs.Textbox(label='Output (string)'),
|
158 |
+
gr.outputs.Textbox(label='Output (raw string)'),
|
159 |
+
gr.outputs.Label(label='Output (label)'),
|
160 |
+
gr.outputs.HTML()
|
161 |
],
|
162 |
examples=[
|
163 |
['miku.jpg',0.5],
|
|
|
168 |
Demo for [KichangKim/DeepDanbooru](https://github.com/KichangKim/DeepDanbooru) with "ready to copy" prompt and a prompt analyzer.
|
169 |
|
170 |
Modified from [hysts/DeepDanbooru](https://huggingface.co/spaces/hysts/DeepDanbooru)
|
171 |
+
|
172 |
PNG Info code forked from [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
173 |
''',
|
174 |
theme=args.theme,
|