Spaces:
Runtime error
Runtime error
Commit
·
eecbd6f
1
Parent(s):
742d08b
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ warnings.filterwarnings('ignore')
|
|
| 24 |
|
| 25 |
import gradio as gr
|
| 26 |
|
|
|
|
| 27 |
|
| 28 |
class GeoTr_Seg(nn.Module):
|
| 29 |
def __init__(self):
|
|
@@ -173,6 +174,6 @@ input_image = gr.inputs.Image()
|
|
| 173 |
output_image = gr.outputs.Image(type='pil')
|
| 174 |
|
| 175 |
|
| 176 |
-
iface = gr.Interface(fn=process_image, inputs=input_image, outputs=output_image, title="DocTr")
|
| 177 |
iface.launch()
|
| 178 |
|
|
|
|
| 24 |
|
| 25 |
import gradio as gr
|
| 26 |
|
| 27 |
+
example_img_list = ['51_1 copy.png','48_2 copy.png','25.jpg']
|
| 28 |
|
| 29 |
class GeoTr_Seg(nn.Module):
|
| 30 |
def __init__(self):
|
|
|
|
| 174 |
output_image = gr.outputs.Image(type='pil')
|
| 175 |
|
| 176 |
|
| 177 |
+
iface = gr.Interface(fn=process_image, inputs=input_image, outputs=output_image, title="DocTr", examples=example_img_list)
|
| 178 |
iface.launch()
|
| 179 |
|