Spaces:
Paused
Paused
Update inference.py
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -47,7 +47,7 @@ def process_document(image):
|
|
| 47 |
|
| 48 |
|
| 49 |
generated_ids = output[0][inputs.input_ids.shape[-1]:]
|
| 50 |
-
print(f"The generated_ids of inference is {generated_ids}")
|
| 51 |
|
| 52 |
# response = processor.decode( # past code
|
| 53 |
# generated_ids,
|
|
@@ -62,7 +62,7 @@ def process_document(image):
|
|
| 62 |
generated_ids,
|
| 63 |
skip_special_tokens=True
|
| 64 |
).strip()
|
| 65 |
-
print(f"The response of inference is {response}")
|
| 66 |
|
| 67 |
# 🔥 FORCE JSON CLEANING
|
| 68 |
start = response.find("{")
|
|
|
|
| 47 |
|
| 48 |
|
| 49 |
generated_ids = output[0][inputs.input_ids.shape[-1]:]
|
| 50 |
+
# print(f"The generated_ids of inference is {generated_ids}")
|
| 51 |
|
| 52 |
# response = processor.decode( # past code
|
| 53 |
# generated_ids,
|
|
|
|
| 62 |
generated_ids,
|
| 63 |
skip_special_tokens=True
|
| 64 |
).strip()
|
| 65 |
+
# print(f"The response of inference is {response}")
|
| 66 |
|
| 67 |
# 🔥 FORCE JSON CLEANING
|
| 68 |
start = response.find("{")
|