Update app.py

app.py

@@ -31,6 +31,9 @@ if uploaded_file is not None:
     for i in range(len(ocr_class)):
         if ocr_class[i].upper() == 'CODE':
             idx.append(ocr_list[i])
-    
 
+
+    st.markdown('**Uploaded Image**')
+    st.image(img, caption='Uploaded Image')
+    st.markdown("**Retrieved Code Block**")
     st.code(("\n").join(idx), language="python", line_numbers=False)