User00100 commited on
Commit
7179595
verified
1 Parent(s): 49b4fb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import streamlit as st
2
  from tensorflow.keras.models import load_model
 
3
  import numpy as np
4
 
5
- # Cargar el modelo Keras
6
- model = load_model('model_m1.h5')
 
7
 
8
  # Funci贸n para preprocesar los datos de entrada
9
  def preprocess_input(values):
 
1
  import streamlit as st
2
  from tensorflow.keras.models import load_model
3
+ from tensorflow.keras import metrics
4
  import numpy as np
5
 
6
+ # Cargar el modelo Keras con la m茅trica mse
7
+ model = load_model('model_m1.h5', custom_objects={'mse': metrics.mean_squared_error})
8
+
9
 
10
  # Funci贸n para preprocesar los datos de entrada
11
  def preprocess_input(values):