More results and adds a tag string output

app.py

@@ -9,6 +9,10 @@ import torchvision.transforms.functional as TVF
 
 
 MODEL_REPO = "fancyfeast/joytag"
+THRESHOLD = 0.4
+DESCRIPTION = """
+Demo for the JoyTag model: https://huggingface.co/fancyfeast/joytag
+"""
 
 
 def prepare_image(image: Image.Image, target_size: int) -> torch.Tensor:
@@ -45,7 +49,11 @@ def predict(image: Image.Image):
 		preds = model(batch)
 		tag_preds = preds['tags'].sigmoid().cpu()
 	
-	return {top_tags[i]: tag_preds[0][i] for i in range(len(top_tags))}
+	scores = {top_tags[i]: tag_preds[0][i] for i in range(len(top_tags))}
+	predicted_tags = [tag for tag, score in scores.items() if score > THRESHOLD]
+	tag_string = ', '.join(predicted_tags)
+
+	return tag_string, scores
 
 
 print("Downloading model...")
@@ -62,8 +70,13 @@ print("Starting server...")
 gradio_app = gr.Interface(
 	predict,
 	inputs=gr.Image(label="Source", sources=['upload', 'webcam'], type='pil'),
-	outputs=[gr.Label(label="Result", num_top_classes=5)],
+	outputs=[
+		gr.Textbox(label="Tag String"),
+		gr.Label(label="Tag Predictions", num_top_classes=100),
+	],
 	title="JoyTag",
+	description=DESCRIPTION,
+	allow_flagging="never",
 )