Update app.py

app.py

@@ -6,7 +6,7 @@ model = joblib.load('decision_tree.pkl')
 
 def onehot(df, column):
     df = df.copy()
-    dummies = pd.get_dummies(df[column])
+    dummies = pd.get_dummies(df[column], prefix='type')
     df = pd.concat([df,dummies], axis=1)
     df = df.drop(column, axis=1)
     return df
@@ -15,7 +15,7 @@ def onehot(df, column):
 def dataframe(file_obj):
     df = pd.read_csv(file_obj.name)
     df = onehot(df, column='type')
-    df = df.drop(['nameOrig','nameDest','isFlaggedFraud'], axis=1)
+    df = df.drop(['nameOrig','nameDest'], axis=1)
     print(df.shape)
     y_pred = model.predict(df)