Update app.py
Browse files
app.py
CHANGED
@@ -39,22 +39,22 @@ def load_repo_dataset(file_path):
|
|
39 |
return pd.read_csv(file_path)
|
40 |
|
41 |
@st.cache_data
|
42 |
-
def load_huggingface_dataset(dataset_name,
|
43 |
dataset = load_dataset(dataset_name, name="all", split="train", trust_remote_code=True, uniform_split=True)
|
44 |
if hasattr(dataset, "to_pandas"):
|
45 |
-
|
46 |
df = dataset.to_pandas()
|
47 |
-
|
48 |
return df
|
49 |
-
|
50 |
return pd.DataFrame(dataset)
|
51 |
|
52 |
@st.cache_data
|
53 |
-
def load_uploaded_csv(uploaded_file,
|
54 |
time.sleep(1) # Simulate processing delay
|
55 |
-
|
56 |
df = pd.read_csv(uploaded_file)
|
57 |
-
|
58 |
return df
|
59 |
|
60 |
# Dataset selection logic
|
|
|
39 |
return pd.read_csv(file_path)
|
40 |
|
41 |
@st.cache_data
|
42 |
+
def load_huggingface_dataset(dataset_name, _progress_callback):
|
43 |
dataset = load_dataset(dataset_name, name="all", split="train", trust_remote_code=True, uniform_split=True)
|
44 |
if hasattr(dataset, "to_pandas"):
|
45 |
+
_progress_callback(50) # Update progress bar
|
46 |
df = dataset.to_pandas()
|
47 |
+
_progress_callback(100) # Final progress update
|
48 |
return df
|
49 |
+
_progress_callback(100)
|
50 |
return pd.DataFrame(dataset)
|
51 |
|
52 |
@st.cache_data
|
53 |
+
def load_uploaded_csv(uploaded_file, _progress_callback):
|
54 |
time.sleep(1) # Simulate processing delay
|
55 |
+
_progress_callback(50) # Update progress bar
|
56 |
df = pd.read_csv(uploaded_file)
|
57 |
+
_progress_callback(100) # Final progress update
|
58 |
return df
|
59 |
|
60 |
# Dataset selection logic
|