DrishtiSharma commited on
Commit
04a79b6
·
verified ·
1 Parent(s): 5987bcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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, 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
 
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