Hugging Face's logo

Spaces:
vishnu23
/
drone_image_segmentation
Runtime error

App Files Community
vishnu23 commited on
Commit
8aefd62
·
1 Parent(s): bfb65a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ import gradio as gr
18
 
19
  def predict(img):
20
  # model = load_model('drive/My Drive/building_footprint_extraction_model.h5')
21
- model = load_model('/content/drive/MyDrive/Colab Notebooks/building_footprint_extraction_model.h5')
22
  img_array = img_to_array(img)
23
  img_array = img_array.reshape((1, 256, 256, 3))
24
  img_array = img_array / 255.0
@@ -30,7 +30,7 @@ def predict(img):
30
 
31
  def get_shape_files(img):
32
  # predictions,_ = predict(img)
33
- model = load_model('/content/drive/MyDrive/Colab Notebooks/building_footprint_extraction_model.h5')
34
  img_array = img_to_array(img)
35
  img_array = img_array.reshape((1, 256, 256, 3))
36
  img_array = img_array / 255.0
 
18
 
19
  def predict(img):
20
  # model = load_model('drive/My Drive/building_footprint_extraction_model.h5')
21
+ model = load_model('building_footprint_extraction_model.h5')
22
  img_array = img_to_array(img)
23
  img_array = img_array.reshape((1, 256, 256, 3))
24
  img_array = img_array / 255.0
 
30
 
31
  def get_shape_files(img):
32
  # predictions,_ = predict(img)
33
+ model = load_model('building_footprint_extraction_model.h5')
34
  img_array = img_to_array(img)
35
  img_array = img_array.reshape((1, 256, 256, 3))
36
  img_array = img_array / 255.0