Hugging Face's logo

Spaces:
Rahatara
/
rag_ColPali_Qwen2VL
Running on Zero

App Files Community
AdrienB134 commited on
Commit
869830f
•
1 Parent(s): 0ea3f72

gre

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def model_inference(
70
 
71
 
72
  prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
73
- inputs = id_processor(text=prompt, images=images, return_tensors="pt")
74
  inputs = {k: v.to("cuda") for k, v in inputs.items()}
75
 
76
  generation_args = {
 
70
 
71
 
72
  prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
73
+ inputs = id_processor(text=prompt, images=[images], return_tensors="pt")
74
  inputs = {k: v.to("cuda") for k, v in inputs.items()}
75
 
76
  generation_args = {