Tzetha commited on
Commit
af75592
·
verified ·
1 Parent(s): 349f574

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -2,7 +2,17 @@ import streamlit as st
2
  import pandas as pd
3
  import numpy as np
4
  import pickle
5
- from sklearn.linear_model import LinearRegression
 
 
 
 
 
 
 
 
 
 
6
  from sklearn.preprocessing import LabelEncoder
7
 
8
  # Load dataset
 
2
  import pandas as pd
3
  import numpy as np
4
  import pickle
5
+ import os
6
+ import subprocess
7
+
8
+ # Ensure scikit-learn is installed
9
+ try:
10
+ import sklearn
11
+ except ImportError:
12
+ subprocess.run(["pip", "install", "scikit-learn", "--quiet"])
13
+ import sklearn
14
+
15
+ from sklearn.linear_model import LinearRegression # Now it should work
16
  from sklearn.preprocessing import LabelEncoder
17
 
18
  # Load dataset