Update app.py

app.py

@@ -1,3 +1,4 @@
+#import all necessary libraries
 import torch
 import numpy as np
 from PIL import Image
@@ -32,7 +33,7 @@ def generate_paragraph(caption):
 
     # Decode the generated output
     generated_text = tokenizer_2.decode(output[0], skip_special_tokens=True)
-    return generated_text
+    return generated_text.capitalize()
 
 # create the Streamlit app
 def app():
@@ -52,8 +53,6 @@ def app():
         string = generate_captions(uploaded_file)
 
         st.image(image, caption='The Uploaded File')
-        st.write("First is first captions for your Photo : ", string)
-
         generated_paragraph = generate_paragraph(string)
 
         st.write(generated_paragraph)