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)