bdsqlsz commited on
Commit
4db78e7
·
verified ·
1 Parent(s): 98a3560

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
 
8
  @spaces.GPU
9
  def create_captions_llava_llama3_docci(image):
10
- pipe = pipeline('open-llava-next-llama3-8b')
11
  gen_config = GenerationConfig(repetition_penalty=1.10)
12
  image = Image.fromarray(np.uint8(image)).convert('RGB')
13
  response = pipe(('Provide a detailed, objective description of the visible content in the images for dataset annotations. Do not include any analysis, interpretations, explanations, or references to the purpose or themes of the image. Focus only on what is visually present', image), gen_config=gen_config)
 
7
 
8
  @spaces.GPU
9
  def create_captions_llava_llama3_docci(image):
10
+ pipe = pipeline('Lin-Chen/open-llava-next-llama3-8b')
11
  gen_config = GenerationConfig(repetition_penalty=1.10)
12
  image = Image.fromarray(np.uint8(image)).convert('RGB')
13
  response = pipe(('Provide a detailed, objective description of the visible content in the images for dataset annotations. Do not include any analysis, interpretations, explanations, or references to the purpose or themes of the image. Focus only on what is visually present', image), gen_config=gen_config)