xzerus commited on
Commit
7d64545
·
verified ·
1 Parent(s): 61b68fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
  # Load the model
7
  model = AutoModel.from_pretrained(
8
- 'OpenGVLab/InternViT-6B-448px-V1-5',
9
  torch_dtype=torch.bfloat16,
10
  low_cpu_mem_usage=True,
11
  trust_remote_code=True,
@@ -13,7 +13,7 @@ model = AutoModel.from_pretrained(
13
  ).cuda().eval()
14
 
15
  # Load the image processor
16
- image_processor = CLIPImageProcessor.from_pretrained('OpenGVLab/InternViT-6B-448px-V1-5')
17
 
18
  # Define the function to process the image and generate outputs
19
  def process_image(image):
 
5
 
6
  # Load the model
7
  model = AutoModel.from_pretrained(
8
+ 'OpenGVLab/InternVL2_5-1B',
9
  torch_dtype=torch.bfloat16,
10
  low_cpu_mem_usage=True,
11
  trust_remote_code=True,
 
13
  ).cuda().eval()
14
 
15
  # Load the image processor
16
+ image_processor = CLIPImageProcessor.from_pretrained('OpenGVLab/InternVL2_5-1B')
17
 
18
  # Define the function to process the image and generate outputs
19
  def process_image(image):