Update app.py

app.py

@@ -14,18 +14,22 @@ from sklearn.model_selection import train_test_split
 x_train, x_test, y_train, y_test = train_test_split(df.drop('charges',axis=1), df['charges'], test_size=0.2, random_state=42)
 
 #decision tree regressor
-from sklearn.tree import DecisionTreeRegressor
-dtree = DecisionTreeRegressor()
+#from sklearn.tree import DecisionTreeRegressor
+#dtree = DecisionTreeRegressor()
 #model training
-dtree.fit(x_train,y_train)
-#model accuracy
-# dtree.score(x_train,y_train)
+#dtree.fit(x_train,y_train)
+
+from sklearn.ensemble import RandomForestRegressor
+rf = RandomForestRegressor()
+#model training
+rf.fit(x_train,y_train)
+
 
 def func(a, b, c, d, e, f):
     
     
     x_test = [[a, b, c, d, e, f]]
-    result = dtree.predict(x_test)[0]
+    result = rf.predict(x_test)[0]
     rounded_result = round(result, 2)
     str="Your Medical Expenses could be: ₹"
     str1="Thankyou🤗 for using our Model"