Update app.py

app.py

@@ -35,8 +35,11 @@ def predict(img, hide: float):
         img = tf.image.per_image_standardization(img)
         data = tf.expand_dims(img, 0)
         out, *_ = model(data)
-    labels = {tag: float(out[i].numpy()) for i, tag in enumerate(tags)}
-    return {k: v for k, v in labels.items() if v >= hide}
+    return {
+        tag: confidence
+        for i, tag in enumerate(tags)
+        if (confidence := float(out[i].numpy())) >= hide
+    }
 
 
 image = inputs.Image(label="Upload your image here!")