yuhangzang commited on
Commit
85c065f
·
1 Parent(s): 1e5cd04
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from PIL import Image
5
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
6
 
7
- MODEL_ID = "internlm/Spatial-SSRL-3B"
8
  MAX_NEW_TOKENS = 2048
9
 
10
  # Example questions from the examples
 
4
  from PIL import Image
5
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
6
 
7
+ MODEL_ID = "internlm/Spatial-SSRL-7B"
8
  MAX_NEW_TOKENS = 2048
9
 
10
  # Example questions from the examples