Update app.py

app.py

@@ -1,11 +1,15 @@
 import gradio as gr
 from seefood_classifier_import import get_seefood_classifier
+import PIL.Image 
+import numpy as np 
 
 classifier = get_seefood_classifier()
 
 def classify(image):
-    print(image)
-    return {str(image): 0.9, 'label2': 0.2}
+    im = PIL.Image.fromarray(np.array(image, dtype=np.uint8))
+    preds = classifier.predict_image(im)
+    results = {pred['label']: pred['score'] for pred in preds}
+    return results
 
 iface = gr.Interface(fn=classify, inputs="image", outputs="label")
 iface.launch()