Update app.py

app.py

@@ -373,6 +373,12 @@ def pneumonia(image):
     colu = list(X_test.columns[selector.get_support()]) #generating name of columns consisting that feature 
     
     Selected_DF = DataFrame[colu]
+    final_model = RandomForestClassifier(criterion = 'gini',max_depth = None,
+                                   max_features = None,max_leaf_nodes = None,
+                                   min_samples_leaf = 3,min_samples_split = 4,
+                                   min_weight_fraction_leaf = 0.0,n_estimators = 200)
+    train_DF = X_train[colu]
+    final_model.fit(train_DF,y_train)
     prediction = model.predict(Selected_DF)
 
     if prediction == 0: # Determine the predicted class