p3nguknight commited on
Commit
495f9d5
·
1 Parent(s): 151727a

Increase duration

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -46,7 +46,7 @@ def image_to_base64(image_path):
46
  return f"data:image/jpeg;base64,{encoded_string}"
47
 
48
 
49
- @spaces.GPU(duration=30)
50
  def pixtral_inference(
51
  images,
52
  text,
@@ -84,7 +84,7 @@ def pixtral_inference(
84
  return result
85
 
86
 
87
- @spaces.GPU(duration=30)
88
  def retrieve(query: str, ds, images, k):
89
  if len(images) == 0:
90
  raise gr.Error("No docs/images for retrieval")
@@ -136,7 +136,7 @@ def convert_files(files):
136
  return images
137
 
138
 
139
- @spaces.GPU(duration=30)
140
  def index_gpu(images, ds):
141
  model = ColPali.from_pretrained(
142
  COLPALI_GEMMA_MODEL_PATH,
 
46
  return f"data:image/jpeg;base64,{encoded_string}"
47
 
48
 
49
+ @spaces.GPU(duration=45)
50
  def pixtral_inference(
51
  images,
52
  text,
 
84
  return result
85
 
86
 
87
+ @spaces.GPU(duration=45)
88
  def retrieve(query: str, ds, images, k):
89
  if len(images) == 0:
90
  raise gr.Error("No docs/images for retrieval")
 
136
  return images
137
 
138
 
139
+ @spaces.GPU(duration=45)
140
  def index_gpu(images, ds):
141
  model = ColPali.from_pretrained(
142
  COLPALI_GEMMA_MODEL_PATH,