Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,7 @@ def layoutjson2md(layout_data: Any) -> str:
|
|
| 140 |
|
| 141 |
|
| 142 |
# --- Core Application Logic ---
|
| 143 |
-
@spaces.GPU
|
| 144 |
def process_document_stream(model_name: str, task_choice: str, image: Image.Image, max_new_tokens: int):
|
| 145 |
"""
|
| 146 |
Main generator function that handles both OCR and Layout Analysis tasks.
|
|
|
|
| 140 |
|
| 141 |
|
| 142 |
# --- Core Application Logic ---
|
| 143 |
+
@spaces.GPU(duration=140) #2min:20secs
|
| 144 |
def process_document_stream(model_name: str, task_choice: str, image: Image.Image, max_new_tokens: int):
|
| 145 |
"""
|
| 146 |
Main generator function that handles both OCR and Layout Analysis tasks.
|