Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,6 @@ def main():
|
|
84 |
if st.sidebar.button('Generate Caption'):
|
85 |
with st.spinner('Generating caption...'):
|
86 |
image_processor, model,tokenizer, device = initialize_model()
|
87 |
-
|
88 |
caption = generate_caption(image_processor, model, device, image)
|
89 |
|
90 |
st.header("Caption:")
|
@@ -97,5 +96,5 @@ if __name__ == '__main__':
|
|
97 |
|
98 |
st.markdown("""
|
99 |
---
|
100 |
-
You are looking at partial
|
101 |
|
|
|
84 |
if st.sidebar.button('Generate Caption'):
|
85 |
with st.spinner('Generating caption...'):
|
86 |
image_processor, model,tokenizer, device = initialize_model()
|
|
|
87 |
caption = generate_caption(image_processor, model, device, image)
|
88 |
|
89 |
st.header("Caption:")
|
|
|
96 |
|
97 |
st.markdown("""
|
98 |
---
|
99 |
+
You are looking at partial tuned model , please JUDGE ME!!! (I am Funny , Sensible , Creative )""")
|
100 |
|