swimmiing commited on
Commit
f7168a8
1 Parent(s): 833fa47

Fix bug in image loader

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -53,7 +53,7 @@ def greet(image, audio):
53
  vt.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)), # CLIP
54
  ])
55
 
56
- image_file = image_transform(image)
57
 
58
  # Inference
59
  placeholder_tokens = model.get_placeholder_token(prompt_template.replace('{}', ''))
@@ -70,8 +70,8 @@ def greet(image, audio):
70
 
71
  return overlaid_image
72
 
73
-
74
- description = 'hello world'
75
 
76
  demo = gr.Interface(
77
  fn=greet,
 
53
  vt.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)), # CLIP
54
  ])
55
 
56
+ image_file = image_transform(image).unsqeeze(0)
57
 
58
  # Inference
59
  placeholder_tokens = model.get_placeholder_token(prompt_template.replace('{}', ''))
 
70
 
71
  return overlaid_image
72
 
73
+ title = 'Audio-Grounded Contrastive Learning (ACL)'
74
+ description = "This is your private demo of our WACV'24 paper Can CLIP Help Sound Source Localization?"
75
 
76
  demo = gr.Interface(
77
  fn=greet,