ChrisMofus commited on
Commit
857f666
1 Parent(s): eb65490

Write the answer

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -22,4 +22,6 @@ if uploaded_file is not None:
22
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
23
 
24
  enc_image = model.encode_image(image)
25
- print(model.answer_question(enc_image, "Describe this image in a few sentences.", tokenizer))
 
 
 
22
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
23
 
24
  enc_image = model.encode_image(image)
25
+ answer = model.answer_question(enc_image, "Describe this image in a few sentences.", tokenizer)
26
+
27
+ st.write(answer)