khang119966 commited on
Commit
dc52088
·
verified ·
1 Parent(s): fe11ab2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,12 +100,12 @@ def load_image(image_file, input_size=448, max_num=12):
100
  return pixel_values
101
 
102
  model = AutoModel.from_pretrained(
103
- "5CD-AI/Vintern-3B-beta",
104
  torch_dtype=torch.bfloat16,
105
  low_cpu_mem_usage=True,
106
  trust_remote_code=True,
107
  ).eval().cuda()
108
- tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Vintern-3B-beta", trust_remote_code=True, use_fast=False)
109
 
110
 
111
  @spaces.GPU
 
100
  return pixel_values
101
 
102
  model = AutoModel.from_pretrained(
103
+ "5CD-AI/Vintern-1B-v3_5",
104
  torch_dtype=torch.bfloat16,
105
  low_cpu_mem_usage=True,
106
  trust_remote_code=True,
107
  ).eval().cuda()
108
+ tokenizer = AutoTokenizer.from_pretrained("5CD-AI/Vintern-1B-v3_5", trust_remote_code=True, use_fast=False)
109
 
110
 
111
  @spaces.GPU