abdulelahagr commited on
Commit
7f99746
·
verified ·
1 Parent(s): f27b541

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,7 +9,9 @@ quantization_config = BitsAndBytesConfig(
9
  model_id = "llava-hf/llava-1.5-7b-hf"
10
 
11
  pipe = pipeline("image-to-text",
12
- model=model_id)
 
 
13
 
14
 
15
  def launch(image, prompt):
 
9
  model_id = "llava-hf/llava-1.5-7b-hf"
10
 
11
  pipe = pipeline("image-to-text",
12
+ model=model_id,
13
+ model_kwargs={"quantization_config": quantization_config}
14
+ )
15
 
16
 
17
  def launch(image, prompt):