nelbarman053 commited on
Commit
d9cbbfa
1 Parent(s): 71abe40

All the necessary things done

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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;">Johan Hagemeyer</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')
 
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
- return caption[0]
 
 
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():