Hugging Face's logo

Spaces:
ybelkada
/
llava-1.5-dlai
Running on Zero

App Files Community
1
ybelkada commited on
Commit
4d23b6a
·
verified ·
1 Parent(s): ef4f6a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
5
 
6
  model_id = "llava-hf/llava-1.5-7b-hf"
7
 
8
- prompt_format = f"USER: <image>\n{}\nASSISTANT:"
9
  image_file = "http://images.cocodataset.org/val2017/000000039769.jpg"
10
 
11
  model = LlavaForConditionalGeneration.from_pretrained(
 
5
 
6
  model_id = "llava-hf/llava-1.5-7b-hf"
7
 
8
+ prompt_format = "USER: <image>\n{}\nASSISTANT:"
9
  image_file = "http://images.cocodataset.org/val2017/000000039769.jpg"
10
 
11
  model = LlavaForConditionalGeneration.from_pretrained(