Update app.py

app.py

@@ -64,18 +64,25 @@ i = 0
 for region in df["Region"].unique():
     df["Region"] = df["Region"].replace(region, str(i))
     i += 1
+
+df = df.astype({"Region": "int"})
     
 i = 0
 
 for country in df["Country"].unique():
     df["Country"] = df["Country"].replace(country, str(i))
     i += 1
-    
-i = 0
 
-for state in df["State"].unique():
-    df["State"] = df["State"].replace(state, str(i))
-    i += 1
+df = df.astype({"Country": "int"})
+
+if merge == False:
+    i = 0
+    
+    for state in df["State"].unique():
+        df["State"] = df["State"].replace(state, str(i))
+        i += 1
+        
+    df = df.astype({"State": "int"})
     
 i = 0
 
@@ -83,9 +90,6 @@ for city in df["City"].unique():
     df["City"] = df["City"].replace(city, str(i))
     i += 1
 
-df = df.astype({"Region": "int"})
-df = df.astype({"Country": "int"})
-df = df.astype({"State": "int"})
 df = df.astype({"City": "int"})
 
 target = 'AvgTemperature'