Update app.py

app.py

@@ -53,6 +53,13 @@ def inference(input_image):
     for i in range(top5_prob.size(0)):
         result[categories[top5_catid[i]]] = top5_prob[i].item()
     print(result)
+    result = {
+    1:0.1,
+    2:0.2,
+    3:0.3,
+    4:0.4,
+    5:0
+    }
     return result
 
 inputs = gr.Image(type='pil')