Pawel_Mar commited on
Commit
3604c28
·
1 Parent(s): 56f0327

fix attempt

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,6 @@ import pandas as pd
5
  import streamlit as st
6
  from ultralytics import YOLO
7
  from streamlit_image_comparison import image_comparison
8
- from ultralytics.utils.ops import scale_image
9
  import numpy as np
10
 
11
  model = YOLO('models/last.pt')
@@ -26,7 +25,8 @@ st.markdown(link, unsafe_allow_html=True)
26
 
27
  def save_uploadedfile(uploadedfile):
28
  with open(os.path.join(parent_media_path, "captured_picture.jpg"), "wb") as f:
29
- f.write(uploadedfile.getbuffer())
 
30
 
31
  def convert_to_jpg(uploaded_image):
32
  im = Image.open(uploaded_image)
@@ -98,7 +98,7 @@ if APPLICATION_MODE == "Take Picture":
98
  ## Function to save image
99
  save_uploadedfile(picture)
100
  st.sidebar.success("Saved File")
101
- captured_picture_img = os.path.join(parent_media_path, "/captured_picture.jpg")
102
  st.write("Click on **Clear photo** to retake picture")
103
  img_file = os.path.join(parent_media_path, "captured_picture.jpg")
104
  st.divider()
 
5
  import streamlit as st
6
  from ultralytics import YOLO
7
  from streamlit_image_comparison import image_comparison
 
8
  import numpy as np
9
 
10
  model = YOLO('models/last.pt')
 
25
 
26
  def save_uploadedfile(uploadedfile):
27
  with open(os.path.join(parent_media_path, "captured_picture.jpg"), "wb") as f:
28
+ im = Image.open(uploadedfile)
29
+ im.save(os.path.join(parent_media_path, "captured_picture.jpg"))
30
 
31
  def convert_to_jpg(uploaded_image):
32
  im = Image.open(uploaded_image)
 
98
  ## Function to save image
99
  save_uploadedfile(picture)
100
  st.sidebar.success("Saved File")
101
+ # captured_picture_img = os.path.join(parent_media_path, "captured_picture.jpg")
102
  st.write("Click on **Clear photo** to retake picture")
103
  img_file = os.path.join(parent_media_path, "captured_picture.jpg")
104
  st.divider()