vumichien commited on
Commit
908df6d
1 Parent(s): 15a3062

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def generate_answer(image, question):
71
  time_start = time.time()
72
  samples = prepare_data(image, question)
73
  samples = model.forward_itm(samples=samples)
74
- samples = model.forward_cap(samples=samples, num_captions=5, num_patches=5)
75
  samples = model.forward_qa_generation(samples)
76
  Img2Prompt = model.prompts_construction(samples)
77
  Img2Prompt_input = tokenizer(Img2Prompt, padding='longest', truncation=True, return_tensors="pt").to(device)
 
71
  time_start = time.time()
72
  samples = prepare_data(image, question)
73
  samples = model.forward_itm(samples=samples)
74
+ samples = model.forward_cap(samples=samples, num_captions=5, num_patches=20)
75
  samples = model.forward_qa_generation(samples)
76
  Img2Prompt = model.prompts_construction(samples)
77
  Img2Prompt_input = tokenizer(Img2Prompt, padding='longest', truncation=True, return_tensors="pt").to(device)