Update app.py

app.py

@@ -15,11 +15,14 @@ from neuralforecast import NeuralForecast
 from neuralforecast.models import NBEATS
 
 REPO_ID = "magcheong/ITI110_Energy_Prediction"
+download_dir = snapshot_download(REPO_ID)
+test_file_path = download_dir +"futr_df.csv"
+test = np.loadtxt(test_file_path, delimiter=",", skiprows=1)
 
 def load_model():
     download_dir = snapshot_download(REPO_ID)
     saved_model_dir = os.path.join(download_dir, "saved_model")
-    test = load_dataset("csv", data_files= "futr_df.csv", data_dir= download_dir)
+    # test = load_dataset("csv", data_files= "futr_df.csv", data_dir= download_dir)
     prediction_model = NeuralForecast.load(saved_model_dir)
     return prediction_model
 
@@ -38,7 +41,7 @@ def UserInputTime(txt):
 
 def predict2(row):
     # test = load_dataset("csv", data_files= "futr_df.csv", data_dir= download_dir)
-    test = DataLoader(dataset=download_dir+"futr_df.csv")
+    # test = DataLoader(dataset=download_dir+"futr_df.csv")
     # prediction = prediction_model.predict(futr_df=test).reset_index()
     # predict_dict = prediction[['ds', 'NBEATS']].to_dict()
     # predicted_value = predict_dict['NBEATS'][row]