jibsn commited on
Commit
2387726
·
verified ·
1 Parent(s): a87bffc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def visualize_molecule(smiles):
49
  def predict(input_image):
50
 
51
  try:
52
- session = ort.InferenceSession("model_v2.onnx")
53
 
54
  img_array,w,h = image_to_numpy(input_image)
55
  processed_image=np.expand_dims(img_array, 0)
 
49
  def predict(input_image):
50
 
51
  try:
52
+ session = ort.InferenceSession("model.onnx")
53
 
54
  img_array,w,h = image_to_numpy(input_image)
55
  processed_image=np.expand_dims(img_array, 0)