Update app.py

app.py

@@ -14,6 +14,13 @@ import pandas as pd
 df_pre = pd.read_csv('data')
 y = df_pre['audience']
 
+filename = 'model.pkl'
+loaded_rf_model = pickle.load(open(filename, 'rb'))
+X_test = pd.DataFrame(
+    data={'saba': [300, 200, 3030, 400], 
+          'date_ymd': [20230328, 20230329, 20230330, 20230331]}
+)
+y_pred_ = loaded_rf_model.predict(X_test, num_iteration=gbm.best_iteration)
 
 def outbreak(date):
   if date:
@@ -21,7 +28,7 @@ def outbreak(date):
     # final_day = date
     # x = np.arange(start_day, final_day + 1)
     fig = plt.figure()
-    plt.plot(y)
+    plt.plot(y_pred_)
     plt.title("audience")
     plt.ylabel("audience")
     plt.xlabel("Days since Day 0")