commit

utils.py

@@ -69,7 +69,7 @@ def make_pred_outside_india(input_img, model, device, user_location):
     pred = F.softmax(pred).detach().numpy()
     y_prob = pred.argmax(axis=1)[0]
 
-    if(user_location=='Outside_India'):
+    if(user_location=='Outside India'):
         class_label = classes_outside_india[y_prob]
     elif(user_location=='India'):
         class_label = classes_india[y_prob]