MissingBreath commited on
Commit
c273644
·
verified ·
1 Parent(s): 8e26a90

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
api.py CHANGED
@@ -135,7 +135,7 @@ loaded_scaler = joblib.load('scaler.joblib')
135
  # Load the saved label encoder
136
  loaded_label_encoder = joblib.load('label_encoder.joblib')
137
 
138
- model = tf.keras.models.load_model('keras_model.h5')
139
 
140
 
141
 
@@ -186,7 +186,7 @@ def predict_crop_disease(input_data):
186
 
187
  print(input_scaled.reshape(1,-1).shape)
188
  # Make predictions
189
- prediction = model.predict(input_scaled)
190
 
191
  # Inverse transform the prediction (if needed, depending on your model)
192
  #predicted_label = loaded_label_encoder.inverse_transform(prediction)
 
135
  # Load the saved label encoder
136
  loaded_label_encoder = joblib.load('label_encoder.joblib')
137
 
138
+ Geomodel = tf.keras.models.load_model('keras_model.h5')
139
 
140
 
141
 
 
186
 
187
  print(input_scaled.reshape(1,-1).shape)
188
  # Make predictions
189
+ prediction = Geomodel.predict(input_scaled)
190
 
191
  # Inverse transform the prediction (if needed, depending on your model)
192
  #predicted_label = loaded_label_encoder.inverse_transform(prediction)