asdf

app.py

@@ -24,6 +24,11 @@ def classify_image(img):
         result[key] = {"color": color_result.get(key, 0.0), "shape": shape_result.get(key, 0.0)}
     return result
 
+def postprocess(prediction):
+    result = {}
+    for key in prediction.keys():
+        result[key] = max(prediction[key].items(), key=lambda x: x[1])[0]
+    return result
 
 image = gr.inputs.Image(shape=(256, 256))
 label = gr.outputs.Label()
@@ -34,8 +39,7 @@ intf = gr.Interface(
     outputs=label, 
     examples="", 
     title="Lego Brick Classifier", 
-    layout="vertical"
+    layout="vertical",
+    postprocess=postprocess
 )
 intf.launch()
-
-