saritha5 commited on
Commit
001ecc6
·
1 Parent(s): b465701

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -29,6 +29,8 @@ from io import BytesIO,StringIO
29
  from pathlib import Path
30
  import h5py
31
 
 
 
32
 
33
  page_names = ["Blurred or Not Blurred Prediction","Occluded or Not Occluded Prediction"]
34
  page = st.sidebar.radio('Navigation',page_names)
@@ -93,8 +95,7 @@ else:
93
  #plt. figure(figsize=(10,9))
94
  def occ_predict(img_content):
95
  im = []
96
- st.write(img_content)
97
- image=cv2.imread(img_content,cv2.COLOR_RGB2YCrCb)
98
  #imgplot = plt.imshow(image)
99
  #plt.show()
100
  img = Image.fromarray(image, 'RGB')
 
29
  from pathlib import Path
30
  import h5py
31
 
32
+ model_file_path = "mobile_net_occ.h5"
33
+
34
 
35
  page_names = ["Blurred or Not Blurred Prediction","Occluded or Not Occluded Prediction"]
36
  page = st.sidebar.radio('Navigation',page_names)
 
95
  #plt. figure(figsize=(10,9))
96
  def occ_predict(img_content):
97
  im = []
98
+ image=cv2.imread(img_content)
 
99
  #imgplot = plt.imshow(image)
100
  #plt.show()
101
  img = Image.fromarray(image, 'RGB')