Upload with huggingface_hub
Browse files- requirements.txt +1 -1
- run.py +2 -2
requirements.txt
CHANGED
@@ -6,4 +6,4 @@ torch==1.6.0+cpu
|
|
6 |
torchvision==0.7.0+cpu
|
7 |
|
8 |
|
9 |
-
https://gradio-main-build.s3.amazonaws.com
|
|
|
6 |
torchvision==0.7.0+cpu
|
7 |
|
8 |
|
9 |
+
https://gradio-main-build.s3.amazonaws.com/a8f961ec7ea4dbb10f8303b9f48d28af83aacbf9/gradio-3.9.1-py3-none-any.whl
|
run.py
CHANGED
@@ -80,8 +80,8 @@ def segment(input):
|
|
80 |
|
81 |
import gradio as gr
|
82 |
|
83 |
-
i = gr.
|
84 |
-
o = gr.
|
85 |
|
86 |
examples = [["img1.jpg"], ["img2.jpg"]]
|
87 |
title = None #"Left Ventricle Segmentation"
|
|
|
80 |
|
81 |
import gradio as gr
|
82 |
|
83 |
+
i = gr.Image(shape=(112, 112), label="Echocardiogram")
|
84 |
+
o = gr.Image(label="Segmentation Mask")
|
85 |
|
86 |
examples = [["img1.jpg"], ["img2.jpg"]]
|
87 |
title = None #"Left Ventricle Segmentation"
|