moldenhof commited on
Commit
d337d91
·
1 Parent(s): 6f0afb9

implementing app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,14 +38,14 @@ dataset = data_cls(data_path="./uploads/")
38
  # dataset.prepare_data()
39
  st.title("Atom Level Entity Detector")
40
 
41
- image_file = st.file_uploader("Upload a chemical structure candidate image",type=['png','jpeg','jpg'])
42
  #st.write('filename is', file_name)
43
  if image_file is not None:
44
  col1, col2 = st.columns(2)
45
 
46
  image = Image.open(image_file)
47
  col1.image(image, use_column_width=True)
48
- with open(os.path.join("uploads/images/",image_file.name),"wb") as f:
49
  f.write(image_file.getbuffer())
50
  st.success("Saved File")
51
  dataset.prepare_data()
 
38
  # dataset.prepare_data()
39
  st.title("Atom Level Entity Detector")
40
 
41
+ image_file = st.file_uploader("Upload a chemical structure candidate image",type=['png'])
42
  #st.write('filename is', file_name)
43
  if image_file is not None:
44
  col1, col2 = st.columns(2)
45
 
46
  image = Image.open(image_file)
47
  col1.image(image, use_column_width=True)
48
+ with open(os.path.join("uploads/images/","0.png"),"wb") as f:
49
  f.write(image_file.getbuffer())
50
  st.success("Saved File")
51
  dataset.prepare_data()