bcvilnrotter commited on
Commit
e290c3b
·
verified ·
1 Parent(s): 1d3826b

Update utils/basic_functions.py

Browse files
Files changed (1) hide show
  1. utils/basic_functions.py +1 -1
utils/basic_functions.py CHANGED
@@ -88,7 +88,7 @@ def huggingface_detect_id_box(model_name,url):
88
  """
89
 
90
  processor,model=load_model(model_name)
91
- inputs = processor(images=image,text=system_prompt,return_tensors="pt").to(model.device)
92
  with torch.no_grad():
93
  output = model.generate(**inputs)
94
 
 
88
  """
89
 
90
  processor,model=load_model(model_name)
91
+ inputs = processor(images=[image],text=system_prompt,return_tensors="pt").to(model.device)
92
  with torch.no_grad():
93
  output = model.generate(**inputs)
94