Update app.py

app.py

@@ -116,7 +116,9 @@ def predict(image: PIL.Image.Image, score_threshold: float,
     
     geninfo = items.get('parameters', geninfo)
     
-    info = ''
+    info = f"""
+<p><h4>PNG Info</h4></p>    
+"""
     for key, text in items.items():
         info += f"""
 <div>
@@ -126,16 +128,14 @@ def predict(image: PIL.Image.Image, score_threshold: float,
 """.strip()+"\n"
     
     if len(info) == 0:
-        message = ""
+        message = "Nothing found in the image."
         info = f"<div><p>{message}<p></div>"
     
-    return (a,info,res)
+    return (info,a,res)
 
 
 def main():
     args = parse_args()
-
-
     model = load_model()
     labels = load_labels()
 
@@ -153,9 +153,9 @@ def main():
                              label='Score Threshold'),
         ],
         [
-            gr.outputs.Textbox(label='Output String'), 
             gr.outputs.HTML(),
-            gr.outputs.Label(label='Output Labels')
+            gr.outputs.Textbox(label='DeepDanbooru Output (string)'), 
+            gr.outputs.Label(label='DeepDanbooru Output (label)')
         ],
         examples=[
         ['miku.jpg',0.5],
@@ -163,8 +163,9 @@ def main():
         ],
         title=TITLE,
         description='''
-Modified from [hysts/DeepDanbooru](https://huggingface.co/spaces/hysts/DeepDanbooru)
+Demo for [KichangKim/DeepDanbooru](https://github.com/KichangKim/DeepDanbooru) with "ready to copy" prompt and a prompt analyzer.
 
+Modified from [hysts/DeepDanbooru](https://huggingface.co/spaces/hysts/DeepDanbooru)
 PNG Info code forked from [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
         ''',
         theme=args.theme,