Update app.py

app.py

@@ -17,6 +17,7 @@ def greet(image):
 
     prediction = np.argmax(prediction,axis=2)
 
+    prediction = np.repeat(prediction[:, :, np.newaxis]*255, 3, axis=2)
     print(prediction.shape)