Update app.py

app.py

@@ -103,6 +103,7 @@ def chatbot(image, question):
     image_features = visual_encoder.encode_images(image.unsqueeze(0))
     emb_mixer.set_image_embeddings(image_features.squeeze(0))
     model.update_emb_weight(emb_mixer.get_input_embeddings())
+    print(emb_mixer.get_input_embeddings().shape)
     print(model.w["emb.weight"].shape)
     image_ids = [i for i in range(emb_mixer.image_start_index, emb_mixer.image_start_index + len(image_features))]
     input_text = generate_prompt(question)