Update app.py
Browse files
app.py
CHANGED
@@ -110,7 +110,7 @@ def imageMode(image, question):
|
|
110 |
inputs = torch.concat((imgToTextEmb, Qtoken_embeddings), axis=-2)
|
111 |
|
112 |
prediction = tokenizer_text.batch_decode(
|
113 |
-
|
114 |
inputs_embeds=inputs,
|
115 |
max_new_tokens=50,
|
116 |
bos_token_id=tokenizer_text.bos_token_id,
|
|
|
110 |
inputs = torch.concat((imgToTextEmb, Qtoken_embeddings), axis=-2)
|
111 |
|
112 |
prediction = tokenizer_text.batch_decode(
|
113 |
+
phi2_finetuned.generate(
|
114 |
inputs_embeds=inputs,
|
115 |
max_new_tokens=50,
|
116 |
bos_token_id=tokenizer_text.bos_token_id,
|