Update app.py

app.py

@@ -18,7 +18,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Load model and processor
-ckpt ="Qwen/Qwen2-VL-7B-Instruct-AWQ"
+ckpt ="Daemontatox/DocumentCogito"
 model = Qwen2VLForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
 
@@ -49,7 +49,7 @@ def process_pdf_file(file_path):
                 if page_text.strip():
                     text += f"Page {page_num + 1}:\n{page_text}\n\n"
                 
-                zoom = 2
+                zoom = 2.5
                 mat = fitz.Matrix(zoom, zoom)
                 pix = page.get_pixmap(matrix=mat, alpha=False)
                 img_data = pix.tobytes("png")
@@ -125,8 +125,8 @@ def process_uploaded_file(file):
         logger.error(f"Error in process_file: {str(e)}")
         return "An error occurred while processing the file. Please try again."
 
-@spaces.GPU(duration=660)
-def bot_streaming(prompt_option, max_new_tokens=8192):
+@spaces.GPU()
+def bot_streaming(prompt_option, max_new_tokens=4096):
     try:
         # Define predetermined prompts
         prompts = {