AkashDataScience commited on
Commit
6473f86
·
1 Parent(s): b52bed7
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,10 +52,10 @@ def infer(message, history):
52
  combined_embeds = []
53
 
54
  with torch.no_grad():
55
- if message['file']:
56
  projected_image_embeds = None
57
  audio_text_embeds = None
58
- for path in message['file']:
59
 
60
  if path.endswith(('.jpg', '.png', '.jpeg')):
61
  image = clip_preprocess(Image.open(path)).unsqueeze(0).to(device)
 
52
  combined_embeds = []
53
 
54
  with torch.no_grad():
55
+ if message['files']:
56
  projected_image_embeds = None
57
  audio_text_embeds = None
58
+ for path in message['files']:
59
 
60
  if path.endswith(('.jpg', '.png', '.jpeg')):
61
  image = clip_preprocess(Image.open(path)).unsqueeze(0).to(device)