Haozhe commited on
Commit
4f54f7b
·
1 Parent(s): cbc410e
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,9 +10,9 @@ import json
10
  import spaces
11
  from serve_constants import html_header, bibtext, learn_more_markdown, tos_markdown
12
 
13
-
14
- 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"
15
- example_image = "/home/ma-user/work/haozhe/workspace/vlspaces/example_images/1.jpg"
16
  # "example_images/document.png"
17
  example_text = "What kind of restaurant is it?"
18
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,
 
10
  import spaces
11
  from serve_constants import html_header, bibtext, learn_more_markdown, tos_markdown
12
 
13
+ MODEL_ID = "TIGER-Lab/PixelReasoner-RL-v1"
14
+ # 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"
15
+ example_image = "example_images/1.jpg" # /home/ma-user/work/haozhe/workspace/vlspaces/
16
  # "example_images/document.png"
17
  example_text = "What kind of restaurant is it?"
18
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,