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():
|