default image change

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.PNG filter=lfs diff=lfs merge=lfs -text
+*.docx filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -6,7 +6,7 @@ from gemini_correction import AiCorrection
 from pathlib import Path
 import os
 
-@st.cache_resource()
+#@st.cache_resource()
 def load_useful():
     return PaddleOCR(lang='en', use_angle_cls=False), AiCorrection()
 
@@ -51,7 +51,7 @@ Then, it sends that text to an AI to clean it up and make it sound better.''')
 
     if use_default_image:
         # If the user chooses default image, load it from a file
-        default_image_path = 'image.jpg'  # Specify the path to your default image
+        default_image_path = 'photo_ocr.jpg'  # Specify the path to your default image
         image = Image.open(default_image_path)
         st.image(image, caption="Default Image", use_container_width=True)
 
@@ -59,12 +59,12 @@ Then, it sends that text to an AI to clean it up and make it sound better.''')
         img_np = np.array(image)
         result = ocr.ocr(img_np, cls=True)
 
-        st.markdown("# Extracted text")
+        st.header("Raw text")
         # Display results
         extracted_text = "\n".join([res[1][0] for line in result for res in line])
         st.write(extracted_text)
-
-        st.markdown("# Corrected version")
+        st.markdown("---")
+        st.header("Corrected version")
         corrected_text = ai_corrector.correct_output(extracted_text)
         st.write(corrected_text)
 
@@ -83,11 +83,14 @@ Then, it sends that text to an AI to clean it up and make it sound better.''')
 
                 
                 extracted_text = "\n".join([res[1][0] for line in result for res in line])
-                st.text_area("Extracted Text", extracted_text, height=200)
+                st.header("Raw text")
+                st.markdown(extracted_text)
 
+            st.markdown("---")
             with st.spinner("Correcting text..."):
                 corrected_text = ai_corrector.correct_output(extracted_text)
-            st.text_area("Corrected Text", corrected_text, height=200)
+            st.header("Corrected Version")
+            st.markdown(corrected_text)
 
 
 if __name__ == "__main__":