Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|