implementing app

app.py

@@ -37,12 +37,15 @@ model_atom.model.roi_heads.score_thresh = 0.65
 
 st.title("Atom Level Entity Detector")
 
-file_name = st.file_uploader("Upload a chemical structure candidate image")
-st.write('filename is', file_name)
-if file_name is not None:
+image_file = st.file_uploader("Upload a chemical structure candidate image",type=['png','jpeg','jpg'])
+#st.write('filename is', file_name)
+if image_file is not None:
    col1, col2 = st.columns(2)
 
-   image = Image.open(file_name)
+   image = Image.open(image_file)
    col1.image(image, use_column_width=True)
+   with open(os.path.join("uploads",image_file.name),"wb") as f: 
+        f.write(image_file.getbuffer())         
+   st.success("Saved File")
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)