Update app.py

app.py

@@ -11,7 +11,6 @@ in_tensor, *_ = inter.get_input_details()
 out_tensor, *_ = inter.get_output_details()
 
 
-@tf.function
 def process_data(content):
     img = tf.io.decode_jpeg(content, channels=3)
     img = tf.image.resize_with_pad(img, SIZE, SIZE, method="nearest")
@@ -26,13 +25,14 @@ def main(file: IO[bytes]):
     inter.set_tensor(in_tensor["index"], data)
     inter.invoke()
     result, *_ = inter.get_tensor(out_tensor["index"])
-    *_, confidence = result
-    print(f"Output confidence: {confidence:.3%}")
-    return confidence
+    safe, questionable, explicit = map(float, result)
+    possibilities = {"safe": safe, "questionable": questionable, "explicit": explicit}
+    print("Predict result:", possibilities)
+    return possibilities
 
 
 image = inputs.Image(type="file")
-label = outputs.Label()
+label = outputs.Label(type="confidences")
 
 interface = Interface(main, image, label)
 interface.launch()