Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -378,7 +378,7 @@ pdf_cache = {
|
|
| 378 |
"is_parsed": False,
|
| 379 |
"results": []
|
| 380 |
}
|
| 381 |
-
@spaces.GPU()
|
| 382 |
def inference(image: Image.Image, prompt: str, max_new_tokens: int = 24000) -> str:
|
| 383 |
"""Run inference on an image with the given prompt"""
|
| 384 |
try:
|
|
|
|
| 378 |
"is_parsed": False,
|
| 379 |
"results": []
|
| 380 |
}
|
| 381 |
+
@spaces.GPU(duration=120)
|
| 382 |
def inference(image: Image.Image, prompt: str, max_new_tokens: int = 24000) -> str:
|
| 383 |
"""Run inference on an image with the given prompt"""
|
| 384 |
try:
|