gokaygokay commited on
Commit
4397550
1 Parent(s): e79a5f3

Update app.py

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