Spaces:
Running
Running
Commit
·
be1ea43
1
Parent(s):
aac970e
Update app.py
Browse files
app.py
CHANGED
@@ -50,8 +50,6 @@ if 'image' in locals(): # Check if image variable exists
|
|
50 |
|
51 |
# Ensure image shape is correct and add batch dimension
|
52 |
img_array = preprocess(image) # This should return an array of shape (1, 256, 256, 3)
|
53 |
-
|
54 |
-
st.info('Worked2')
|
55 |
|
56 |
predicted_class, confidence = classify_garbage(img_array, model)
|
57 |
|
@@ -62,5 +60,4 @@ if 'image' in locals(): # Check if image variable exists
|
|
62 |
st.info('Worked4')
|
63 |
|
64 |
except Exception as e:
|
65 |
-
st.info('Error')
|
66 |
st.error(f"An error occurred: {e}")
|
|
|
50 |
|
51 |
# Ensure image shape is correct and add batch dimension
|
52 |
img_array = preprocess(image) # This should return an array of shape (1, 256, 256, 3)
|
|
|
|
|
53 |
|
54 |
predicted_class, confidence = classify_garbage(img_array, model)
|
55 |
|
|
|
60 |
st.info('Worked4')
|
61 |
|
62 |
except Exception as e:
|
|
|
63 |
st.error(f"An error occurred: {e}")
|