jayebaku commited on
Commit
f988ad7
·
verified ·
1 Parent(s): 7e77357

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -10,11 +10,12 @@ from statistics import mean
10
  HFTOKEN = os.environ["HF_TOKEN"]
11
 
12
  def load_and_classify_csv(file, text_field, event_model):
13
- # if ".csv" in file.name:
14
- # df = pd.read_csv(file.name)
15
- # else ".tsv" in file.name:
16
- df = pd.read_table(file.name)
17
- print(file.name)
 
18
 
19
  if text_field not in df.columns:
20
  raise gr.Error(f"Error: Enter text column'{text_field}' not in CSV file.")
 
10
  HFTOKEN = os.environ["HF_TOKEN"]
11
 
12
  def load_and_classify_csv(file, text_field, event_model):
13
+ filepath = file.name
14
+ if ".csv" in filepath:
15
+ df = pd.read_csv(filepath)
16
+ else ".tsv" in filepath:
17
+ df = pd.read_table(filepath)
18
+ # print(file.name)
19
 
20
  if text_field not in df.columns:
21
  raise gr.Error(f"Error: Enter text column'{text_field}' not in CSV file.")