rjiang12 commited on
Commit
1d036b5
·
1 Parent(s): b265742

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -47,7 +47,7 @@ def generate_answer_git(processor, model, image, question):
47
  print(generated_ids)
48
  generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
49
  print(generated_answer)
50
- generated_answer1 = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)[1]
51
  print(generated_answer1)
52
 
53
  return 'haha'
@@ -56,10 +56,14 @@ def generate_answer_git(processor, model, image, question):
56
  def generate_answer_blip(processor, model, image, question):
57
  # prepare image + question
58
  inputs = processor(images=image, text=question, return_tensors="pt")
59
-
60
  generated_ids = model.generate(**inputs, max_length=50, output_scores=True)
61
  print(generated_ids)
62
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
 
 
 
 
63
 
64
  return generated_answer
65
 
 
47
  print(generated_ids)
48
  generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
49
  print(generated_answer)
50
+ generated_answer1 = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)[0]
51
  print(generated_answer1)
52
 
53
  return 'haha'
 
56
  def generate_answer_blip(processor, model, image, question):
57
  # prepare image + question
58
  inputs = processor(images=image, text=question, return_tensors="pt")
59
+ print('blip')
60
  generated_ids = model.generate(**inputs, max_length=50, output_scores=True)
61
  print(generated_ids)
62
  generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
63
+ generated_answer0 = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)[0]
64
+ print(generated_answer0)
65
+ generated_answer1 = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)[1]
66
+ print(generated_answer1)
67
 
68
  return generated_answer
69