Singularity666 commited on
Commit
bc93555
·
1 Parent(s): ffaac97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def download_link(content, filename, link_text):
61
  href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
62
  return href
63
 
64
- def show_predicted_caption(image, top_k=8):
65
  matches = predict_caption(
66
  image, model, text_embeddings, testing_df["caption"]
67
  )[:top_k]
 
61
  href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
62
  return href
63
 
64
+ def show_predicted_caption(image, top_k=1):
65
  matches = predict_caption(
66
  image, model, text_embeddings, testing_df["caption"]
67
  )[:top_k]