anirudh-valyx commited on
Commit
0474ce4
·
1 Parent(s): 5bca57b

requirements

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ def answer_question(image, question):
18
  end = outputs.end_logits.argmax(-1).item() + 1
19
 
20
  answer = processor.tokenizer.decode(inputs["input_ids"][0][start:end])
 
21
  return answer
22
 
23
  iface = gr.Interface(
 
18
  end = outputs.end_logits.argmax(-1).item() + 1
19
 
20
  answer = processor.tokenizer.decode(inputs["input_ids"][0][start:end])
21
+
22
  return answer
23
 
24
  iface = gr.Interface(