Update label.py

label.py

@@ -99,7 +99,7 @@ def load_model():
     return model
 
 
-def predict_environment(image):
+def predict_environment(img):
     # load the labels
     classes, labels_IO, labels_attribute, W_attribute = load_labels()
     
@@ -115,7 +115,7 @@ def predict_environment(image):
     weight_softmax = params[-2].data.numpy()
     weight_softmax[weight_softmax<0] = 0
 
-    img = Image.open(image)
+    #img = Image.open(image)
     input_img = V(tf(img).unsqueeze(0))
     
     # forward pass