Update app.py

app.py

@@ -10,12 +10,13 @@ model = tf.saved_model.load(model_path)
 classes = [ "bleached" ,  "healthy" , ]
 
 def run(image_path):
-    img = Image.open(image_path).convert('RGB')
+    img = Image.open(i).convert('RGB')
     img = img.resize((300, 300 * img.size[1] // img.size[0]), Image.ANTIALIAS)
-    inp_numpy = numpy.array(img)[None]
-    inp = tensorflow.constant(inp_numpy, dtype='float32')
+    inp_numpy = np.array(img)[None]
+    inp = tf.constant(inp_numpy, dtype='float32')
     class_scores = model(inp)[0].numpy()
-    return class_scores
+    state = classes[class_scores.argmax()]
+    return state
 
 title = "Trash Detector"
 description = (
@@ -28,6 +29,7 @@ interface = gr.Interface(
     run,
     inputs=[gr.components.Image(type="filepath")],
     outputs="text",
+    #outputs=gradio.outputs.Label(num_top_classes=3),
     title=title,
     description=description,
     examples=examples,