Update app.py

app.py

@@ -40,7 +40,7 @@ def calculate_md5_from_binary(binary_data):
     hash_md5.update(binary_data)
     return hash_md5.hexdigest()
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=90)
 def add_pdf_gradio(pdf_file_binary, progress=gr.Progress()):
     global model, tokenizer
 
@@ -84,7 +84,6 @@ def add_pdf_gradio(pdf_file_binary, progress=gr.Progress()):
     
     return knowledge_base_name
 
-# @spaces.GPU
 def retrieve_gradio(knowledge_base: str, query: str, topk: int):
     global model, tokenizer
 
@@ -138,6 +137,8 @@ model.to(device)
 
 with gr.Blocks() as app:
     gr.Markdown("# Memex: OCR-free Visual Document Retrieval @RhapsodyAI [RhapsodyAI/minicpm-visual-embedding-v0](https://huggingface.co/RhapsodyAI/minicpm-visual-embedding-v0)")
+
+    gr.Markdown("Only PDF file is available now. You are suggested to uplaod <= 50 pages per PDF, otherwise you will reach GPU time limit.")
     
     with gr.Row():
         file_input = gr.File(type="binary", label="Upload PDF")