update

app.py

@@ -13,7 +13,7 @@ from serve_constants import html_header, bibtext, learn_more_markdown, tos_markd
 MODEL_ID = "TIGER-Lab/PixelReasoner-RL-v1"
 # MODEL_ID = "/home/ma-user/work/haozhe/workspace/lmm-r1/toolckpts/pix17K0506wt-NormalizedPenalizedFixedReweightCont-256-lossvernone-samplevernone-fmtnone-group-n8-ml10000-lr10-sysvcot-8node/global_step24_hf_evalbest"  
 example_image = "example_images/1.jpg" # /home/ma-user/work/haozhe/workspace/vlspaces/
-# "example_images/document.png"
+# example_image = "/home/ma-user/work/haozhe/workspace/vlspaces/example_images/1.jpg"
 example_text = "What kind of restaurant is it?"
 processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,
                                           #   min_pixels=min_pixels, 
@@ -160,6 +160,15 @@ def model_inference(input_dict, history):
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
     
+    current_path = os.getcwd()
+    print(f"Current running path: {current_path}")
+
+    # Define the folder name to check
+    folder_to_find = "example_images"
+
+    # Create the full path to the folder
+    folder_path = os.path.join(current_path, folder_to_find)
+    print('files', files)
     imagelist = rawimagelist = current_message_images = [load_image(image) for image in files]
     all_images += current_message_images
     messages.append({