Spaces:
Sleeping
Sleeping
Pawel_Mar commited on
Commit ·
3604c28
1
Parent(s): 56f0327
fix attempt
Browse files
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 |
-
|
|
|
|
| 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, "
|
| 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()
|