Spaces:

AdrienB134
/

rag_colpali_idefics3

Build error

AdrienB134 commited on Aug 30, 2024

Commit

c785d73

1 Parent(s): b728f0b

ruyj

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ def model_inference(
     prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
-    inputs = id_processor(text=prompt, images=[images], return_tensors="pt")
     inputs = {k: v.to("cuda") for k, v in inputs.items()}
     generation_args = {

     prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
+    inputs = id_processor(text=prompt, images=images, return_tensors="pt")
     inputs = {k: v.to("cuda") for k, v in inputs.items()}
     generation_args = {