AI-ANK commited on
Commit
f21d103
·
1 Parent(s): 30bf3a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -23,6 +23,8 @@ def get_vision_model():
23
  processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
24
  return model, processor
25
 
 
 
26
  # Function to get image caption via Kosmos2.
27
  @st.cache_data
28
  def get_image_caption(image_data):
 
23
  processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
24
  return model, processor
25
 
26
+ model, processor = get_vision_model()
27
+
28
  # Function to get image caption via Kosmos2.
29
  @st.cache_data
30
  def get_image_caption(image_data):