comparing-captioning-models

Running

nielsr HF staff commited on Jul 13, 2023

Commit

5dd3d52

•

1 Parent(s): 740fb26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def generate_caption_blip2(processor, model, image, replace_token=False):
     inputs = processor(images=image, text=prompt, return_tensors="pt").to(device=model.device, dtype=torch.float16)
-    generated_ids = model.generate(pixel_values=inputs.pixel_values,
                                    num_beams=5, max_length=50, min_length=1, top_p=0.9,
                                    repetition_penalty=1.5, length_penalty=1.0, temperature=1)
     if replace_token:

     inputs = processor(images=image, text=prompt, return_tensors="pt").to(device=model.device, dtype=torch.float16)
+    generated_ids = model.generate(**inputs,
                                    num_beams=5, max_length=50, min_length=1, top_p=0.9,
                                    repetition_penalty=1.5, length_penalty=1.0, temperature=1)
     if replace_token: