wjbmattingly commited on
Commit
75478dd
·
verified ·
1 Parent(s): 5d256ae

Update app.py

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