Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -352,7 +352,7 @@ pdf_cache = {
|
|
352 |
"is_parsed": False,
|
353 |
"results": []
|
354 |
}
|
355 |
-
@spaces.GPU()
|
356 |
def inference(image: Image.Image, prompt: str, max_new_tokens: int = 24000) -> str:
|
357 |
"""Run inference on an image with the given prompt"""
|
358 |
try:
|
|
|
352 |
"is_parsed": False,
|
353 |
"results": []
|
354 |
}
|
355 |
+
@spaces.GPU(duration=300)
|
356 |
def inference(image: Image.Image, prompt: str, max_new_tokens: int = 24000) -> str:
|
357 |
"""Run inference on an image with the given prompt"""
|
358 |
try:
|