lambertxiao commited on
Commit
36cbd96
·
verified ·
1 Parent(s): b98d7a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from PIL import Image
4
  import torch
5
  import numpy as np
6
 
7
- model_name_or_path = "lyttt/VLV_captioner"
8
- model = AutoModel.from_pretrained(model_name_or_path, revision="master", trust_remote_code=True,low_cpu_mem_usage=False)
9
 
10
  # @spaces.GPU(duration=120)
11
  def greet(image):
 
4
  import torch
5
  import numpy as np
6
 
7
+ model_name_or_path = "lambertxiao/Vision-Language-Vision-Captioner-Qwen2.5-3B"
8
+ model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True, low_cpu_mem_usage=False)
9
 
10
  # @spaces.GPU(duration=120)
11
  def greet(image):