Update app.py

app.py

@@ -38,7 +38,7 @@ def inference(sample_name):
     # Remove any singleton dimensions if necessary
     image_data = np.moveaxis(image_data,0,-1)
     image = Image.fromarray(image_data)
-    return age, predicted_age[0][0], image
+    return age, round(predicted_age[0][0]), image
 
 # List of sample file names
 sample_files = sorted(os.listdir(os.path.join(os.getcwd(),'samples')))