gokaygokay commited on
Commit
307a01e
1 Parent(s): 4397550

Update app.py

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