Update app.py

app.py

@@ -16,9 +16,11 @@ def classify_text(text):
     
     return model.config.id2label[predicted_class_id]
 
-uploaded_file = st.file_uploader("Upload Image", type= ['png', 'jpg']) 
+uploaded_file = st.file_uploader("Upload Image", type= ['png', 'jpeg', 'jpg']) 
 
 if uploaded_file is not None:
+    st.image(uploaded_file)
+    print(type(uploaded_file))
     ocr_list = [x for x in pytesseract.image_to_string(uploaded_file).split("\n") if x != '']
     ocr_class = [classify_text(x) for x in ocr_list]
     idx = []