Spaces:
Running
Running
v3 to v4.
Browse files
app.py
CHANGED
|
@@ -2,7 +2,6 @@ import gradio as gr
|
|
| 2 |
import torch
|
| 3 |
from visual_clutter import Vlc
|
| 4 |
|
| 5 |
-
|
| 6 |
def inference(img):
|
| 7 |
clt = Vlc(img.name, numlevels=3, contrast_filt_sigma=1, contrast_pool_sigma=3, color_pool_sigma=3, prefix='test')
|
| 8 |
# get Feature Congestion clutter of a test map:
|
|
@@ -14,16 +13,16 @@ def inference(img):
|
|
| 14 |
title = 'Visual Clutter'
|
| 15 |
description = 'Compute two measures of visual clutter (Feature Congestion and Subband Entropy), see the code at https://github.com/kargaranamir/visual-clutter'
|
| 16 |
article = "<p style='text-align: center'></p>"
|
| 17 |
-
examples = [['test.jpg'],['test2.jpg']]
|
| 18 |
css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
|
| 19 |
|
| 20 |
gr.Interface(
|
| 21 |
inference,
|
| 22 |
-
[gr.
|
| 23 |
-
[gr.
|
| 24 |
title=title,
|
| 25 |
description=description,
|
| 26 |
article=article,
|
| 27 |
examples=examples,
|
| 28 |
css=css,
|
| 29 |
-
|
|
|
|
| 2 |
import torch
|
| 3 |
from visual_clutter import Vlc
|
| 4 |
|
|
|
|
| 5 |
def inference(img):
|
| 6 |
clt = Vlc(img.name, numlevels=3, contrast_filt_sigma=1, contrast_pool_sigma=3, color_pool_sigma=3, prefix='test')
|
| 7 |
# get Feature Congestion clutter of a test map:
|
|
|
|
| 13 |
title = 'Visual Clutter'
|
| 14 |
description = 'Compute two measures of visual clutter (Feature Congestion and Subband Entropy), see the code at https://github.com/kargaranamir/visual-clutter'
|
| 15 |
article = "<p style='text-align: center'></p>"
|
| 16 |
+
examples = [['test.jpg'], ['test2.jpg']]
|
| 17 |
css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
|
| 18 |
|
| 19 |
gr.Interface(
|
| 20 |
inference,
|
| 21 |
+
[gr.Image(type='file', label='Input')],
|
| 22 |
+
[gr.Image(type='file', label='Feature Congestion Output Image'), gr.Textbox(label="Feature Congestion"), gr.Textbox(label="Subband Entropy")],
|
| 23 |
title=title,
|
| 24 |
description=description,
|
| 25 |
article=article,
|
| 26 |
examples=examples,
|
| 27 |
css=css,
|
| 28 |
+
).launch(debug=True, enable_queue=True)
|