Spaces:
Runtime error
Runtime error
Commit
·
14fb0af
1
Parent(s):
7ade7eb
Update app.py
Browse files
app.py
CHANGED
|
@@ -151,8 +151,8 @@ with app:
|
|
| 151 |
with gr.Row():
|
| 152 |
with gr.Column():
|
| 153 |
input_video = gr.Image(type='filepath')
|
| 154 |
-
model_id = gr.Dropdown(value=model_path[0], choices=model_path)
|
| 155 |
-
cca = gr.Dropdown(value="Connected Components Labelling", choices=["Connected Components Labelling","No Post Process"])
|
| 156 |
|
| 157 |
input_video_button = gr.Button(value="Predict")
|
| 158 |
|
|
@@ -163,7 +163,7 @@ with app:
|
|
| 163 |
output_mask_image = gr.Image(type='filepath')
|
| 164 |
|
| 165 |
|
| 166 |
-
gr.Examples(examples, inputs=[input_video, model_id], outputs=[output_orijinal_image, output_mask_image], fn=Segformer_Segmentation, cache_examples=True)
|
| 167 |
input_video_button.click(Segformer_Segmentation, inputs=[input_video, model_id,cca], outputs=[output_orijinal_image, output_mask_image])
|
| 168 |
|
| 169 |
app.launch(debug=True)
|
|
|
|
| 151 |
with gr.Row():
|
| 152 |
with gr.Column():
|
| 153 |
input_video = gr.Image(type='filepath')
|
| 154 |
+
model_id = gr.Dropdown(value=model_path[0], choices=model_path,label="Model Name")
|
| 155 |
+
cca = gr.Dropdown(value="Connected Components Labelling", choices=["Connected Components Labelling","No Post Process"],label="Post Process")
|
| 156 |
|
| 157 |
input_video_button = gr.Button(value="Predict")
|
| 158 |
|
|
|
|
| 163 |
output_mask_image = gr.Image(type='filepath')
|
| 164 |
|
| 165 |
|
| 166 |
+
gr.Examples(examples, inputs=[input_video, model_id,"Connected Components Labelling"], outputs=[output_orijinal_image, output_mask_image], fn=Segformer_Segmentation, cache_examples=True)
|
| 167 |
input_video_button.click(Segformer_Segmentation, inputs=[input_video, model_id,cca], outputs=[output_orijinal_image, output_mask_image])
|
| 168 |
|
| 169 |
app.launch(debug=True)
|