update app.py

app.py

@@ -1,4 +1,5 @@
 import traceback
+import logging
 from typing import Optional
 import spaces
 import gradio as gr
@@ -21,6 +22,9 @@ import torch
 from transformers import AutoModelForCausalLM
 from transformers import AutoProcessor 
 
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
 # Define repository and local directory
 repo_id = "microsoft/OmniParser-v2.0"  # HF repo
 local_dir = "weights"  # Target local directory
@@ -159,6 +163,7 @@ def process(
     instruction,
 ) -> Optional[Image.Image]:
 
+    logger.info("Starting processing.")
     try:
         # image_save_path = 'imgs/saved_image_demo.png'
         # image_input.save(image_save_path)
@@ -250,13 +255,15 @@ def process(
                 )
             except:
                 image_som = image_input
-        
+
+        logger.info("finish processing")
         return image_som, str(parsed_content_list)
     except Exception as e:
-        print('error in process')
+        logger.info('error in process')
         traceback.print_exc()
         return image_input, 'error in process'
 
+logger.info("Starting App.")
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)
     with gr.Row():