rohan-16 commited on
Commit
fc5c9f1
·
verified ·
1 Parent(s): ded0b3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -364,11 +364,11 @@ elif selected_menu == "Prediction":
364
 
365
  # Load the selected model
366
  if selected_model == "Model 1":
367
- model_path = '/content/drive/MyDrive/output_1/emotion_model.h5'
368
- history = np.load('/content/drive/MyDrive/output_1/saved_history_1.npz', allow_pickle=True)
369
  else:
370
- model_path = '/content/drive/MyDrive/output_1/emotion_model_new1234.h5'
371
- history = np.load('/content/drive/MyDrive/output_1/saved_history_2.npz', allow_pickle=True)
372
 
373
  model = load_model(model_path)
374
  # Extract the training and validation loss from the history
 
364
 
365
  # Load the selected model
366
  if selected_model == "Model 1":
367
+ model_path = 'output/emotion_model.h5'
368
+ history = np.load('output/saved_history_1.npz', allow_pickle=True)
369
  else:
370
+ model_path = 'output/emotion_model_new1234.h5'
371
+ history = np.load('output/saved_history_2.npz', allow_pickle=True)
372
 
373
  model = load_model(model_path)
374
  # Extract the training and validation loss from the history