Revert "Return a list of predicted labels"

This reverts commit 72d940356accfca18261d089a669501e0c0ff778.

app.py

@@ -7,7 +7,6 @@ import functools
 import os
 import pathlib
 import tarfile
-import tempfile
 
 import deepdanbooru as dd
 import gradio as gr
@@ -65,8 +64,7 @@ def load_labels() -> list[str]:
 
 
 def predict(image: PIL.Image.Image, score_threshold: float,
-            model: tf.keras.Model,
-            labels: list[str]) -> tuple[dict[str, float], str]:
+            model: tf.keras.Model, labels: list[str]) -> dict[str, float]:
     _, height, width, _ = model.input_shape
     image = np.asarray(image)
     image = tf.image.resize(image,
@@ -83,14 +81,7 @@ def predict(image: PIL.Image.Image, score_threshold: float,
         if prob < score_threshold:
             continue
         res[label] = prob
-
-    sorted_preds = sorted(res.items(), key=lambda x: -x[1])
-    out_path = tempfile.NamedTemporaryFile(suffix='.txt', delete=False)
-    with open(out_path.name, 'w') as f:
-        for key, _ in sorted_preds:
-            f.write(f'{key}\n')
-
-    return res, out_path.name
+    return res
 
 
 def main():
@@ -115,10 +106,7 @@ def main():
                       value=args.score_threshold,
                       label='Score Threshold'),
         ],
-        [
-            gr.Label(label='Output'),
-            gr.File(label='Tag List'),
-        ],
+        gr.Label(label='Output'),
         examples=examples,
         title=TITLE,
         description=DESCRIPTION,