LPDoctor commited on
Commit
bd5c61e
1 Parent(s): d86bacf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -8,8 +8,6 @@ from transformers import BitsAndBytesConfig, pipeline
8
  import re
9
  import time
10
 
11
- DESCRIPTION = "# LLaVA 🌋"
12
-
13
  model_id = "llava-hf/llava-1.5-7b-hf"
14
  quantization_config = BitsAndBytesConfig(
15
  load_in_4bit=True,
@@ -42,7 +40,7 @@ def infer(image, prompt,
42
  max_length,
43
  min_length,
44
  top_p):
45
-
46
  outputs = pipe(images=image, prompt=prompt,
47
  generate_kwargs={"temperature":temperature,
48
  "length_penalty":length_penalty,
 
8
  import re
9
  import time
10
 
 
 
11
  model_id = "llava-hf/llava-1.5-7b-hf"
12
  quantization_config = BitsAndBytesConfig(
13
  load_in_4bit=True,
 
40
  max_length,
41
  min_length,
42
  top_p):
43
+ print("类型是:",type(image))
44
  outputs = pipe(images=image, prompt=prompt,
45
  generate_kwargs={"temperature":temperature,
46
  "length_penalty":length_penalty,