swimmiing commited on
Commit
8095871
·
1 Parent(s): ddf85f7

Add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. 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?'\n"
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