Update app.py

app.py

@@ -52,8 +52,12 @@ def predict_tags(image_url, threshold=0.5):
 def gradio_predict(url, threshold):
     tags, error = predict_tags(url, threshold)
     if error:
-        return error
-    return "\n".join([f"{tag}: {score:.2f}" for tag, score in tags])
+        return {"error": error}
+    if not tags:
+        return {"error": "No tags above threshold."}
+    
+    top_tag, top_score = tags[0]
+    return {"tag_name": top_tag, "tag_score": round(top_score, 4)}
 
 demo = gr.Interface(
     fn=gradio_predict,