VanguardAI commited on
Commit
fc8841b
·
verified ·
1 Parent(s): 1fd9e99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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: