Beasto commited on
Commit
d67d24b
·
1 Parent(s): b67aed8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -11,8 +11,14 @@ f = st.file_uploader("Choose a Photo")
11
  # Read the video file from the file-like object
12
  if f is not None:
13
  img = Image.open(f)
 
 
 
14
  img = img.resize((256,256))
15
  img = np.reshape(img,(1,256,256,3))
 
 
 
16
  pred,pred2 = model.predict(img),model2.predict(img)
17
 
18
  st.image(img,use_column_width=True)
 
11
  # Read the video file from the file-like object
12
  if f is not None:
13
  img = Image.open(f)
14
+
15
+ img2 = img.resize((200,200))
16
+
17
  img = img.resize((256,256))
18
  img = np.reshape(img,(1,256,256,3))
19
+
20
+ img2 = np.reshape(img2,(1,200,200,3))
21
+
22
  pred,pred2 = model.predict(img),model2.predict(img)
23
 
24
  st.image(img,use_column_width=True)