Update app.py

app.py

@@ -27,6 +27,11 @@ df = pd.read_csv("city_temperature.csv")
 
 country = "all"
 
+if merge == True:
+    df.loc[df['State'].notna(), 'Country'] = df['State']
+    df = df.loc[:, ~df.columns.str.contains('State')]
+   
+
 if all_region == False:
     region = st.selectbox(
          'Which region do you want to predict temparature ?',
@@ -56,10 +61,6 @@ if all_city == False:
     
     
 
-def mergeStateToCountry():
-    df.loc[df['State'].notna(), 'Country'] = df['State']
-    df = df.loc[:, ~df.columns.str.contains('State')]
-
 i = 0
 
 for region in df["Region"].unique():