bndl commited on
Commit
413f5b2
·
1 Parent(s): acd6f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -167,7 +167,8 @@ def train_model(x, target_cols):
167
  print("Selected target columns")
168
  print(target_cols)
169
  time.sleep(6)
170
- performance_plot = cv2.imread("model_performance.png")
 
171
  metrics = pd.DataFrame([[0.05, 0.017]], columns=["RMSE", "MAPE"])
172
  next_df = x.sample(n=5, random_state=12)
173
  next_df.drop(columns=target_cols, inplace=True)
 
167
  print("Selected target columns")
168
  print(target_cols)
169
  time.sleep(6)
170
+ # performance_plot = cv2.imread("model_performance.png")
171
+ performance_plot = cv2.imread("predictions_ground_truth.png")
172
  metrics = pd.DataFrame([[0.05, 0.017]], columns=["RMSE", "MAPE"])
173
  next_df = x.sample(n=5, random_state=12)
174
  next_df.drop(columns=target_cols, inplace=True)