VishnuEcoClim commited on
Commit
be1ea43
·
1 Parent(s): aac970e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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}")