Spaces:

ToletiSri
/

Capstone

Sleeping

ToletiSri commited on Jan 28

Commit

7907b26

•

1 Parent(s): a704279

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ def imageMode(image, question):
     inputs = torch.concat((imgToTextEmb, Qtoken_embeddings), axis=-2)
     prediction = tokenizer_text.batch_decode(
-        phi2.generate(
             inputs_embeds=inputs,
             max_new_tokens=50,
             bos_token_id=tokenizer_text.bos_token_id,

     inputs = torch.concat((imgToTextEmb, Qtoken_embeddings), axis=-2)
     prediction = tokenizer_text.batch_decode(
+        phi2_finetuned.generate(
             inputs_embeds=inputs,
             max_new_tokens=50,
             bos_token_id=tokenizer_text.bos_token_id,