Update app.py

app.py

@@ -100,9 +100,17 @@ def get_predicted_attributes(image_urls, category):
     # Clean up the results into one long string
     for i, result in enumerate(common_result):
         common_result[i] = ", ".join([f"{x[0]}" for x in result])
+    
+    result = {}
 
+    # Iterate through the list and split each item into key and value
+    for item in common_result:
+        # Split by ': ' to separate the key and value
+        key, value = item.split(': ', 1)
+        # Add to the dictionary
+        result[key] = value
 
-    return common_result
+    return result