nelbarman053
commited on
Commit
·
d9cbbfa
1
Parent(s):
71abe40
All the necessary things done
Browse files- app.py +1 -1
- pages/Caption_Generator.py +3 -1
app.py
CHANGED
@@ -35,7 +35,7 @@ def main():
|
|
35 |
|
36 |
st.markdown("##### 3. Boom!! Your imaginary caption is ready.")
|
37 |
|
38 |
-
st.markdown(f'<div style="display: flex; justify-content: center;"><p style="font-size: 25px; font-weight: bold; color: blue;">
|
39 |
|
40 |
if st.button(label="Go To Generator", type="primary"):
|
41 |
st.switch_page(page='pages/Caption_Generator.py')
|
|
|
35 |
|
36 |
st.markdown("##### 3. Boom!! Your imaginary caption is ready.")
|
37 |
|
38 |
+
st.markdown(f'<div style="display: flex; justify-content: center;"><p style="font-size: 25px; font-weight: bold; color: blue;">Portrait Artist Son</p></div>', unsafe_allow_html = True)
|
39 |
|
40 |
if st.button(label="Go To Generator", type="primary"):
|
41 |
st.switch_page(page='pages/Caption_Generator.py')
|
pages/Caption_Generator.py
CHANGED
@@ -73,7 +73,9 @@ def get_caption(_processor, _tokenizer, _model, image):
|
|
73 |
skip_special_tokens = True
|
74 |
)
|
75 |
|
76 |
-
|
|
|
|
|
77 |
|
78 |
# Displaying elements
|
79 |
def DisplayInteractionElements():
|
|
|
73 |
skip_special_tokens = True
|
74 |
)
|
75 |
|
76 |
+
caption = " ".join([item[0].upper()+item[1:] for item in caption[0].split(" ")])
|
77 |
+
|
78 |
+
return caption
|
79 |
|
80 |
# Displaying elements
|
81 |
def DisplayInteractionElements():
|