Update app.py

app.py

@@ -43,7 +43,7 @@ def reduce_image_count(image):
         for j in range(output_size[1]):
             block = image[i*block_size[0]:(i+1)*block_size[0], j*block_size[1]:(j+1)*block_size[1]]
             count = np.count_nonzero(block)
-            output[i, j] = 1 - ((count / (block_size[0] * block_size[1])) * 1)
+            output[i, j] = 16 - ((count / (block_size[0] * block_size[1])) * 1)
 
     return output
 
@@ -64,6 +64,6 @@ def predict(img):
     otpt=u
     #pred,pred_idx,probs = learn.predict(img)
     
-    return dict([[labels[i], float(otpt[i])] for i in range(len(labels))]),inp_img
+    return dict([[labels[i], float(otpt[i])] for i in range(len(labels))]),inp_img/16
 
 gr.Interface(fn=predict, inputs=gr.inputs.Image(type='filepath'),   outputs=[gr.outputs.Label(num_top_classes=10), gr.outputs.Image()]).launch()