jableable commited on
Commit
a3b1bd8
1 Parent(s): c1963d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -6,6 +6,16 @@ import keras
6
 
7
  loaded_model = keras.saving.load_model("best_model.keras")
8
 
 
 
 
 
 
 
 
 
 
 
9
  #model = from_pretrained_keras("jableable/road_model")
10
 
11
  #pipe = pipeline('sentiment-analysis')
 
6
 
7
  loaded_model = keras.saving.load_model("best_model.keras")
8
 
9
+ uploaded_img = st.file_uploader("Upload your file here...",type=['png', 'jpeg', 'jpg'])
10
+
11
+ if uploaded_file is not None:
12
+ st.image(uploaded_file)
13
+ array = np.array(uploaded_img)
14
+ result = loaded_model.predict(array)
15
+
16
+ st.write(f"Your prediction is: {result}")
17
+
18
+
19
  #model = from_pretrained_keras("jableable/road_model")
20
 
21
  #pipe = pipeline('sentiment-analysis')