Update app.py

app.py

@@ -20,8 +20,13 @@ def load_model(model_choice):
         with open('dtreeweights.pkl', 'rb') as pickleFile:
             return pickle.load(pickleFile)
     elif model_choice == "Sequential Model":
-        with open('my_seq_model (1).pkl', 'rb') as pickleFile:
-            return joblib.load(pickleFile)
+        try:
+            # Try loading the Sequential model saved using joblib
+            with open('my_seq_model (1).pkl', 'rb') as pickleFile:
+                return joblib.load(pickleFile)
+        except:
+            # If joblib loading fails, fallback to TensorFlow loading
+            return load_model('my_seq_model (1)')
     else:
         raise ValueError("Invalid model selection")