Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|