Update app.py

app.py

@@ -17,11 +17,16 @@ import PIL.Image
 import tensorflow as tf
 import piexif
 import piexif.helper
+import re
 
 TITLE = 'DeepDanbooru String'
 
+url = 'https://huggingface.co/spaces/yo2266911/DeepDanbooru_string/'
+repo_id, repo_type = re.findall(r'https://huggingface.co/([^/]+)/([^/]+)/', url)
+print(f'Repository ID: {repo_id}')
+print(f'Repository type: {repo_type}')
+MODEL_REPO = f'{repo_id}/{repo_type}'
 TOKEN = os.environ['TOKEN']
-MODEL_REPO = 'yo2266911/DeepDanbooru_string'
 MODEL_FILENAME = 'model-resnet_custom_v3.h5'
 LABEL_FILENAME = 'tags.txt'
 
@@ -95,7 +100,6 @@ def predict(image: PIL.Image.Image, score_threshold: float,
         res[label] = prob
     b = dict(sorted(res.items(),key=lambda item:item[1], reverse=True))
     a = ', '.join(list(b.keys())).replace('_',' ').replace('(','\(').replace(')','\)')
-    c = ', '.join(list(b.keys()))
     
     items = rawimage.info
     geninfo = ''
@@ -132,7 +136,7 @@ def predict(image: PIL.Image.Image, score_threshold: float,
         message = "Nothing found in the image."
         info = f"<div><p>{message}<p></div>"
     
-    return (a,c,res,info)
+    return (info,a,res)
 
 
 def main():
@@ -154,10 +158,9 @@ def main():
                              label='Score Threshold'),
         ],
         [
-            gr.outputs.Textbox(label='Output (string)'), 
-            gr.outputs.Textbox(label='Output (raw string)'), 
-            gr.outputs.Label(label='Output (label)'),
-            gr.outputs.HTML()
+            gr.outputs.HTML(),
+            gr.outputs.Textbox(label='DeepDanbooru Output (string)'), 
+            gr.outputs.Label(label='DeepDanbooru Output (label)')
         ],
         examples=[
         ['miku.jpg',0.5],
@@ -168,7 +171,6 @@ def main():
 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,