Update label.py

label.py

@@ -115,7 +115,7 @@ weight_softmax = params[-2].data.numpy()
 weight_softmax[weight_softmax<0] = 0
 
 def predict(img):
-    img = Image.open('6.jpg')
+    #img = Image.open('6.jpg')
     input_img = V(tf(img).unsqueeze(0))
     logit = model.forward(input_img)
     h_x = F.softmax(logit, 1).data.squeeze()