Sher1988 commited on
Commit
90c960d
·
verified ·
1 Parent(s): 67d0f13

Update src/image_classifier_app.py

Browse files
Files changed (1) hide show
  1. src/image_classifier_app.py +7 -13
src/image_classifier_app.py CHANGED
@@ -1,16 +1,10 @@
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())
 
1
+ import streamlit as st
2
+
3
+ st.title("Upload Test")
4
 
5
+ uploaded_file = st.file_uploader("Choose a file")
 
 
 
6
 
7
+ st.write("uploaded_file value:", uploaded_file)
 
 
8
 
9
+ if uploaded_file:
10
+ st.success("File detected")