Update app.py

app.py

@@ -109,13 +109,13 @@ def imageMode(image, question):
     print('-------question embedding from phi2 obtained-----------')
     inputs = torch.concat((imgToTextEmb, Qtoken_embeddings), axis=-2)
     
-    prediction = tokenizer.batch_decode(
+    prediction = tokenizer_text.batch_decode(
         phi2.generate(
             inputs_embeds=inputs, 
             max_new_tokens=50,
-            bos_token_id=tokenizer.bos_token_id, 
-            eos_token_id=tokenizer.eos_token_id,
-            pad_token_id=tokenizer.pad_token_id
+            bos_token_id=tokenizer_text.bos_token_id, 
+            eos_token_id=tokenizer_text.eos_token_id,
+            pad_token_id=tokenizer_text.pad_token_id
         )
     )
     text_pred = prediction[0].rstrip('<|endoftext|>').rstrip("\n")