wenhu commited on
Commit
31d3f3e
·
verified ·
1 Parent(s): b5465a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,9 +11,11 @@ import spaces
11
  import os
12
  from serve_constants import html_header, bibtext, learn_more_markdown, tos_markdown
13
 
 
 
14
  MODEL_ID = "TIGER-Lab/PixelReasoner-RL-v1"
15
  # 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"
16
- example_image = "example_images/1.jpg" # /home/ma-user/work/haozhe/workspace/vlspaces/
17
  # example_image = "/home/ma-user/work/haozhe/workspace/vlspaces/example_images/1.jpg"
18
  example_text = "What kind of restaurant is it?"
19
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,
 
11
  import os
12
  from serve_constants import html_header, bibtext, learn_more_markdown, tos_markdown
13
 
14
+ cur_dir = os.path.dirname(os.path.abspath(__file__))
15
+
16
  MODEL_ID = "TIGER-Lab/PixelReasoner-RL-v1"
17
  # 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"
18
+ example_image = f"{cur_dir}/example_images/1.jpg" # /home/ma-user/work/haozhe/workspace/vlspaces/
19
  # example_image = "/home/ma-user/work/haozhe/workspace/vlspaces/example_images/1.jpg"
20
  example_text = "What kind of restaurant is it?"
21
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,