Update app.py

app.py

@@ -79,6 +79,13 @@ def tag_image(pil_image: Image.Image, output_format: str) -> str:
     else:  # Detailed output
         if not results_by_cat:
             return "No tags predicted for this image."
+
+        # Ensure artist tag in detailed output even if below threshold
+        if 'artist' not in results_by_cat and all_artist_tags_probs:
+            best_artist_tag, best_artist_prob = max(all_artist_tags_probs, key=lambda item: item[1])
+            results_by_cat['artist'] = [(best_artist_tag, best_artist_prob)]
+
+
         lines = []
         lines.append("**Predicted Tags by Category:**  \n")  # (Markdown newline: two spaces + newline)
         for cat, tag_list in results_by_cat.items():