GotThatData commited on
Commit
e2cfe8a
·
verified ·
1 Parent(s): 3fbda65
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -3,14 +3,19 @@ from pydrive2.drive import GoogleDrive
3
  import os
4
  import gradio as gr
5
  from datasets import load_dataset, Dataset, concatenate_datasets
6
- from huggingface_hub import create_repo, upload_file, login
7
- from pathlib import Path
8
  import pandas as pd
9
  from PIL import Image
10
  from tqdm import tqdm
11
  import logging
12
  import yaml
13
- import json
 
 
 
 
 
 
 
14
 
15
  # Set up logging
16
  logging.basicConfig(
 
3
  import os
4
  import gradio as gr
5
  from datasets import load_dataset, Dataset, concatenate_datasets
 
 
6
  import pandas as pd
7
  from PIL import Image
8
  from tqdm import tqdm
9
  import logging
10
  import yaml
11
+ from huggingface_hub import login # Add this import
12
+
13
+ # Authenticate with Hugging Face
14
+ HF_TOKEN = os.getenv('HF_TOKEN')
15
+ if HF_TOKEN:
16
+ login(token=HF_TOKEN)
17
+ else:
18
+ logger.warning("No Hugging Face token found. Please add HF_TOKEN to your Space secrets.")
19
 
20
  # Set up logging
21
  logging.basicConfig(