Update app.py

app.py

@@ -9,14 +9,24 @@ import os
 print(f"Prediction environment scikit-learn version: {sklearn.__version__}")
 
 def decode_file(file_path):
-    with open(file_path, 'rb') as file:
-        obj = pickle.load(file)
-    return obj
+    try:
+        with open(file_path, 'rb') as file:
+            obj = pickle.load(file)
+        return obj
+    except FileNotFoundError:
+        print(f"Error: File not found at {file_path}")
+        return None
+    except Exception as e:
+        print(f"Error loading pickle file: {e}")
+        return None
 
 # Load the model once when starting the app
 try:
-    model = decode_file('model.pkl')
-    print("Model loaded successfully")
+    model = decode_file('./model.pkl') # Updated here
+    if model is not None:
+        print("Model loaded successfully")
+    else:
+        print("Model loading failed, None value")
 except Exception as e:
     print(f"Error loading model: {e}")
     model = None
@@ -107,7 +117,7 @@ def predict(gender, age, hypertension, ever_married, work_type, heart_disease,
         prediction = model.predict_proba(processed_input)[0][1]
         return f"The probability of stroke is {prediction:.2%}"
     except Exception as e:
-        return f"Error making prediction: {str(e)}"
+        return f"Error making prediction: {str(e)}, model is not valid" # updated here
 
 # Create the Gradio interface
 iface = gr.Interface(