BrandDead commited on
Commit
312451f
1 Parent(s): 7bc2bf9

Update paths to models directory

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,10 +5,10 @@ import numpy as np
5
  import joblib
6
 
7
  # Load models
8
- rnn_model = load_model('/content/drive/My Drive/virgil_rnn_model.keras', compile=False)
9
- gan_generator = load_model('/content/drive/My Drive/virgil_gan_generator.keras', compile=False)
10
- vae_model = load_model('/content/drive/MyDrive/virgil_autoencoder_model.keras', compile=False)
11
- rf_model = joblib.load('/content/drive/MyDrive/virgil_rf_finetuned_model.pkl')
12
 
13
  # Define functions for each model
14
  def learn_fashion(input_data):
 
5
  import joblib
6
 
7
  # Load models
8
+ rnn_model = load_model('models/virgil_rnn_model.keras', compile=False)
9
+ gan_generator = load_model('models/virgil_gan_generator.keras', compile=False)
10
+ vae_model = load_model('models/virgil_autoencoder_model.keras', compile=False)
11
+ rf_model = joblib.load('models/virgil_rf_finetuned_model.pkl')
12
 
13
  # Define functions for each model
14
  def learn_fashion(input_data):