Update app.py

app.py

@@ -1,3 +1,13 @@
 import gradio as gr
 
-gr.Interface.load("models/ongkn/attraction-classifier").launch()
+def predict_and_label(inp):
+    print("Warning: you need to extract a face and resize it to (224, 224) before accurate results can be achieved!")
+    iface = gr.Interface.load("models/ongkn/attraction-classifier")
+    result = iface.predict(inp)
+    # Assuming result is a dictionary with class labels as keys and probabilities as values
+    max_label = max(result, key=result.get)
+    print(max_label)
+    return result
+
+iface = gr.Interface(fn=predict_and_label, inputs=gr.inputs.Image(), outputs="label")
+iface.launch()