Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ def load_model(model_choice):
|
|
| 21 |
return pickle.load(pickleFile)
|
| 22 |
elif model_choice == "Sequential Model":
|
| 23 |
with open('my_seq_model (1).pkl', 'rb') as pickleFile:
|
| 24 |
-
return
|
| 25 |
else:
|
| 26 |
raise ValueError("Invalid model selection")
|
| 27 |
|
|
|
|
| 21 |
return pickle.load(pickleFile)
|
| 22 |
elif model_choice == "Sequential Model":
|
| 23 |
with open('my_seq_model (1).pkl', 'rb') as pickleFile:
|
| 24 |
+
return joblib.load(pickleFile)
|
| 25 |
else:
|
| 26 |
raise ValueError("Invalid model selection")
|
| 27 |
|