Update app.py

app.py

@@ -74,8 +74,13 @@ def main():
                                                "three_g": [Has_3G],
                                                "touch_screen": [Has_touch_screen],
                                                "wifi": [Has_wifi]}))
-            #result = '>50k' if result[0] == 1 else '<50k'
-            st.success(f'Your predicted Mobile prices is {result[0]}')
+            result = if result[0] == 0:
+            result = 'low cost' 
+            elif result[0] == 1:
+            result = 'high cost' 
+            else: 
+            result = 'very high cost'
+            st.success('Your predicted Mobile prices is'+result)
 if __name__ == "__main__":
     main()
                        # Show prediction