Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|