updated result

app.py

@@ -25,11 +25,7 @@ def model_prediction(path):
     with st.spinner('Model is being loaded..'):
         model=load_model()
     yhat = model.predict(np.expand_dims(resize/255, 0))
-    if(yhat>0.5):
-        result = "Prediction is loose"
-    else:
-        result = "Prediction is control"
-    return result
+    return yhat
 
 def on_click():
     if file is None:
@@ -39,10 +35,9 @@ def on_click():
         st.image(image, use_column_width=True)
         image = image.convert('RGB')
         predictions = model_prediction(np.array(image))
-        st.write(predictions)
-        print(
-            "This image most likely belongs to {}."
-            .format(predictions)
-)
+        if (predictions>0.5):
+            st.write("""# Prediction : Implant is loose""")
+        else:
+            st.write("""# Prediction : Implant is in control""")
 
 st.button('Predict', on_click=on_click)