snajmark commited on
Commit
cc6fe26
·
1 Parent(s): 222e647

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def predict(x, request: gr.Request):
77
  """
78
  Predict the hardness and yield strength using the ML model. Input data is a dataframe
79
  """
80
- loaded_model = tf.keras.models.load_model("hardness.h5")
81
  print("summary is", loaded_model.summary())
82
  #x = x.replace("", 0)
83
  x = np.asarray(x).astype("float32")
 
77
  """
78
  Predict the hardness and yield strength using the ML model. Input data is a dataframe
79
  """
80
+ loaded_model = tf.keras.models.load_model("hardness_nn_graph_separate_elements.h5")
81
  print("summary is", loaded_model.summary())
82
  #x = x.replace("", 0)
83
  x = np.asarray(x).astype("float32")