Spaces:
Running
Running
Add requirements.txt
Browse files- app.py +3 -3
- requirements.txt +2 -1
app.py
CHANGED
@@ -72,10 +72,9 @@ def greet(image, audio):
|
|
72 |
|
73 |
|
74 |
title = "Zero-shot sound source localization with ACL"
|
75 |
-
description = "This is simple demo of our WACV'24 paper 'Can CLIP Help Sound Source Localization?'\
|
76 |
-
description += "To use it, simply upload an image and corresponding audio to mask (identify in the image)"
|
77 |
-
description += "or use one of the examples below and click 'submit'. Results will show up in a few seconds."
|
78 |
|
|
|
79 |
|
80 |
demo = gr.Interface(
|
81 |
fn=greet,
|
@@ -83,6 +82,7 @@ demo = gr.Interface(
|
|
83 |
outputs=gr.Image(type="pil"),
|
84 |
title=title,
|
85 |
description=description,
|
|
|
86 |
)
|
87 |
|
88 |
demo.launch(debug=True)
|
|
|
72 |
|
73 |
|
74 |
title = "Zero-shot sound source localization with ACL"
|
75 |
+
description = "This is simple demo of our WACV'24 paper 'Can CLIP Help Sound Source Localization?'\nTo use it, simply upload an image and corresponding audio to mask (identify in the image) or use one of the examples below and click 'submit'. Results will show up in a few seconds."
|
|
|
|
|
76 |
|
77 |
+
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2311.04066'>Can CLIP Help Sound Source Localization?</a> | <a href='https://github.com/swimmiing/ACL-SSL'>Official Github Repository</a></p>"
|
78 |
|
79 |
demo = gr.Interface(
|
80 |
fn=greet,
|
|
|
82 |
outputs=gr.Image(type="pil"),
|
83 |
title=title,
|
84 |
description=description,
|
85 |
+
article=article,
|
86 |
)
|
87 |
|
88 |
demo.launch(debug=True)
|
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch==1.13
|
|
2 |
torchvision==0.14.0
|
3 |
torchaudio==0.13.0
|
4 |
transformers==4.25.1
|
5 |
-
Pillow
|
|
|
|
2 |
torchvision==0.14.0
|
3 |
torchaudio==0.13.0
|
4 |
transformers==4.25.1
|
5 |
+
Pillow
|
6 |
+
opencv-python
|