kadirnar commited on
Commit
2040dd8
1 Parent(s): 4995fb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -4,6 +4,7 @@ import matplotlib.pyplot as plt
4
 
5
  import gradio as gr
6
 
 
7
 
8
  def keras_inference(img_data, model_path):
9
  model_path = attempt_download_from_hub(model_path)
@@ -17,15 +18,15 @@ def keras_inference(img_data, model_path):
17
 
18
  inputs = [
19
  gr.Image(type='filepath', label='Image'),
20
- gr.Dropdown(['kadirnar/Keras-Segmenting-Buildings-v1'], value="kadirnar/Keras-Segmenting-Buildings-v1", label='Model Path')
21
  ]
22
 
23
  outputs = gr.Image(label='Segmentation')
24
 
25
  examples = [
26
- ['data/testv1.jpg', 'kadirnar/Keras-Segmenting-Buildings-v1'],
27
- ['data/testv2.jpg', 'kadirnar/Keras-Segmenting-Buildings-v1'],
28
- ['data/testv3.jpg', 'kadirnar/Keras-Segmenting-Buildings-v1'],
29
  ]
30
 
31
  title = 'Segmenting Buildings in Satellite Images with Keras'
 
4
 
5
  import gradio as gr
6
 
7
+ model_path = 'deprem-ml/deprem-keras-satellite-semantic-mapping'
8
 
9
  def keras_inference(img_data, model_path):
10
  model_path = attempt_download_from_hub(model_path)
 
18
 
19
  inputs = [
20
  gr.Image(type='filepath', label='Image'),
21
+ gr.Dropdown([model_path], value=model_path, label='Model Path')
22
  ]
23
 
24
  outputs = gr.Image(label='Segmentation')
25
 
26
  examples = [
27
+ ['data/testv1.jpg', model_path],
28
+ ['data/testv2.jpg', model_path],
29
+ ['data/testv3.jpg', model_path],
30
  ]
31
 
32
  title = 'Segmenting Buildings in Satellite Images with Keras'