Vr commited on
Commit
bb0a118
·
1 Parent(s): 9ffe799
Deployment/app.py → app.py RENAMED
@@ -48,7 +48,7 @@ try:
48
  if st.button('Predict'):
49
  img = preprocess(image)
50
  model = model_arc()
51
- model.load_weights("../weights/model.h5")
52
  prediction = model.predict(img[np.newaxis, ...])
53
  proba = np.max(prediction[0], axis=-1)
54
  print("Probability:",np.max(prediction[0], axis=-1))
 
48
  if st.button('Predict'):
49
  img = preprocess(image)
50
  model = model_arc()
51
+ model.load_weights("weights/model.h5")
52
  prediction = model.predict(img[np.newaxis, ...])
53
  proba = np.max(prediction[0], axis=-1)
54
  print("Probability:",np.max(prediction[0], axis=-1))
Deployment/requirements.txt → requirements.txt RENAMED
File without changes
Deployment/utils.py → utils.py RENAMED
File without changes