Spaces:

swimmiing
/

ACL-SSL-zeroshot-demo

Running

swimmiing commited on Dec 13, 2023

Commit

6f752a6

•

1 Parent(s): f7168a8

Fix bug in image loader v2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def greet(image, audio):
         vt.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)),  # CLIP
     ])
-    image_file = image_transform(image).unsqeeze(0)
     # Inference
     placeholder_tokens = model.get_placeholder_token(prompt_template.replace('{}', ''))
@@ -77,7 +77,7 @@ demo = gr.Interface(
     fn=greet,
     inputs=[gr.Image(type='pil'), gr.Audio()],
     outputs=gr.Image(type="pil"),
-    title='AudioToken',
     description=description,
 )

         vt.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)),  # CLIP
     ])
+    image_file = image_transform(image).unsqueeze(0)
     # Inference
     placeholder_tokens = model.get_placeholder_token(prompt_template.replace('{}', ''))
     fn=greet,
     inputs=[gr.Image(type='pil'), gr.Audio()],
     outputs=gr.Image(type="pil"),
+    title=title,
     description=description,
 )