Spaces:
Sleeping
Sleeping
vishalkatheriya18
commited on
Commit
•
3500d9f
1
Parent(s):
dacd259
Update app.py
Browse files
app.py
CHANGED
@@ -40,9 +40,9 @@ if url:
|
|
40 |
st.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
|
41 |
|
42 |
start_time = time.time()
|
43 |
-
|
44 |
# Predict using YOLO
|
45 |
-
predictions = predict_with_yolo(
|
46 |
|
47 |
# Display predictions
|
48 |
if predictions:
|
|
|
40 |
st.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
|
41 |
|
42 |
start_time = time.time()
|
43 |
+
image_resized = image.resize((224, 224))
|
44 |
# Predict using YOLO
|
45 |
+
predictions = predict_with_yolo(image_resized)
|
46 |
|
47 |
# Display predictions
|
48 |
if predictions:
|