Sher1988 commited on
Commit
67d0f13
·
verified ·
1 Parent(s): 74362cb

Create image_classifier_app.py

Browse files
Files changed (1) hide show
  1. src/image_classifier_app.py +16 -0
src/image_classifier_app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ if uploaded_file:
2
+ st.write("Inside upload block")
3
+
4
+ try:
5
+ st.write("Opening image")
6
+ image = Image.open(uploaded_file).convert("RGB")
7
+ st.write("Image opened")
8
+
9
+ st.write("Calling load_model()")
10
+ model, device = load_model()
11
+ st.write("Model loaded successfully")
12
+
13
+ except Exception as e:
14
+ import traceback
15
+ st.error("Crash detected:")
16
+ st.text(traceback.format_exc())