Spaces:
Running
on
Zero
Running
on
Zero
gokaygokay
commited on
Commit
•
4397550
1
Parent(s):
e79a5f3
Update app.py
Browse files
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)
|