commit

app.py

@@ -19,7 +19,7 @@ model_V1 = 'indian_efficientnet_b0.pt'
 @st.cache()
 def model_prediction(model_path, img_file, rescale,selected_location):
     input_img, device = load_prepare_img(img_file)
-    if(selected_location=='Outside_India'):
+    if(selected_location=='Outside India'):
         model = getmodel_outside_india(model_path)
         prediction = make_pred_outside_india(input_img, model, device, selected_location)
     elif(selected_location=='India'):
@@ -46,7 +46,7 @@ def main():
     with col1: 
         # image uploading button
         uploaded_file = st.file_uploader("Choose a file")
-        selected_location = st.selectbox('Select loaction',('India', 'Outside_India'), index=1)
+        selected_location = st.selectbox('Select loaction',('India', 'Outside India'), index=1)
         if uploaded_file is not None:
             display_img = uploaded_file.read()
             uploaded_img = Image.open(uploaded_file)
@@ -74,7 +74,7 @@ def main():
                     if selected_location == 'India':
                         pred_model = model_V1 
                         pred_rescale = True
-                    if selected_location == 'Outside_India':
+                    if selected_location == 'Outside India':
                         pred_model = model_V2
                         pred_rescale =True