rjiang12 commited on
Commit
fe6ca74
1 Parent(s): bbec7cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -38,8 +38,8 @@ def generate_answer_git(processor, model, image, question):
38
  input_ids = [processor.tokenizer.cls_token_id] + input_ids
39
  input_ids = torch.tensor(input_ids).unsqueeze(0)
40
 
41
- generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50)
42
- print(generated_ids.logits)
43
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
44
 
45
  return generated_answer
@@ -49,8 +49,8 @@ def generate_answer_blip(processor, model, image, question):
49
  # prepare image + question
50
  inputs = processor(images=image, text=question, return_tensors="pt")
51
 
52
- generated_ids = model.generate(**inputs, max_length=50)
53
- print(generated_ids.logits)
54
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
55
 
56
  return generated_answer
 
38
  input_ids = [processor.tokenizer.cls_token_id] + input_ids
39
  input_ids = torch.tensor(input_ids).unsqueeze(0)
40
 
41
+ generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, output_scores=True, return_dict_in_generate=True)
42
+ print(generated_ids)
43
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
44
 
45
  return generated_answer
 
49
  # prepare image + question
50
  inputs = processor(images=image, text=question, return_tensors="pt")
51
 
52
+ generated_ids = model.generate(**inputs, max_length=50, output_scores=True, return_dict_in_generate=True)
53
+ print(generated_ids)
54
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
55
 
56
  return generated_answer