Update app.py

app.py

@@ -256,8 +256,6 @@ def predict(image):
     std = np.expand_dims(std, axis=(1,2))
     
     image = (image / 255.0 - mean)/std
-
-    print("normalized image:", image.shape)
     
     # Convert the image to a numpy array and add a batch dimension
     if len(input_shape) == 4 and input_shape[0] == 1:
@@ -268,9 +266,11 @@ def predict(image):
     # Perform inference
     output = model.run(None, {input_name: image})
     # print(type(output))
-    print("model.run worked")
+    print(output)
+
+    annotated_img = output[0]
 
-    return output
+    return annotated_img
     
 # Gradio interface
 demo = gr.Interface(