gokaygokay commited on
Commit
e79a5f3
·
verified ·
1 Parent(s): efb77d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,10 +1,9 @@
1
  from lmdeploy import pipeline
2
  from lmdeploy.vl import load_image
3
 
4
- pipe = pipeline('gokaygokay/llava-llama3-docci')
5
-
6
  @spaces.GPU
7
  def create_captions_llava_llama3_docci(image):
 
8
  gen_config = GenerationConfig(repetition_penalty=1.10)
9
  image = Image.fromarray(np.uint8(image)).convert('RGB')
10
  response = pipe(('describe this image in detail', image), gen_config=gen_config)
 
1
  from lmdeploy import pipeline
2
  from lmdeploy.vl import load_image
3
 
 
 
4
  @spaces.GPU
5
  def create_captions_llava_llama3_docci(image):
6
+ pipe = pipeline('gokaygokay/llava-llama3-docci')
7
  gen_config = GenerationConfig(repetition_penalty=1.10)
8
  image = Image.fromarray(np.uint8(image)).convert('RGB')
9
  response = pipe(('describe this image in detail', image), gen_config=gen_config)