Spaces:
Sleeping
Sleeping
Update utils/basic_functions.py
Browse files- 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(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(image=image,text=system_prompt,return_tensors="pt").to(model.device)
|
| 92 |
with torch.no_grad():
|
| 93 |
output = model.generate(**inputs)
|
| 94 |
|