Hugging Face's logo

Spaces:
Vinit710
/
road_omniview
Building

App Files Community
Vinit710 commited on
Commit
18c0b24
·
verified ·
1 Parent(s): faacc87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import cv2
5
 
6
  # Load your model (convert .h5 path to model)
7
- MODEL_PATH = "model.h5" # put your .h5 in the same directory as app.py
8
  model = tf.keras.models.load_model(MODEL_PATH)
9
 
10
  def predict_road_mask(image):
 
4
  import cv2
5
 
6
  # Load your model (convert .h5 path to model)
7
+ MODEL_PATH = "road_detection_resnet_e50.h5" # put your .h5 in the same directory as app.py
8
  model = tf.keras.models.load_model(MODEL_PATH)
9
 
10
  def predict_road_mask(image):