fjh

app.py

@@ -96,7 +96,10 @@ def index_gpu(images, ds):
  embeddings_doc = model(**batch_doc)
  ds.extend(list(torch.unbind(embeddings_doc.to("cpu"))))
  return f"Uploaded and converted {len(images)} pages", ds, images
-
+ 
+@spaces.GPU
+def answer_gpu():
+ return 0
 
 def get_example():
  return [[["climate_youth_magazine.pdf"], "How much tropical forest is cut annually ?"]]
@@ -133,7 +136,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
  search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[output_gallery])
 
  answer_button = gr.Button("Answer", variant="primary")
- answer_button.click(answer, inputs=[])
+ answer_button.click(answer_gpu, inputs=[])
 
 if __name__ == "__main__":
  demo.queue(max_size=10).launch(debug=True)