Vasudevakrishna commited on
Commit
7eb436c
1 Parent(s): 7e28176

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def generate_answers(img=None, aud = None, q = None, max_tokens = 30):
65
  audio_embeds = phi2_model.model.embed_tokens(audio_tokens.to(config.get("device")))
66
  inputs_embeddings.append(audio_embeds)
67
 
68
- if q is not None:
69
  ques = tokenizer(q, return_tensors="pt", return_attention_mask=False)['input_ids']
70
  q_embeds = phi2_model.model.embed_tokens(ques.to(config.get("device")))
71
  inputs_embeddings.append(q_embeds)
 
65
  audio_embeds = phi2_model.model.embed_tokens(audio_tokens.to(config.get("device")))
66
  inputs_embeddings.append(audio_embeds)
67
 
68
+ if q!='':
69
  ques = tokenizer(q, return_tensors="pt", return_attention_mask=False)['input_ids']
70
  q_embeds = phi2_model.model.embed_tokens(ques.to(config.get("device")))
71
  inputs_embeddings.append(q_embeds)