Spaces:
Sleeping
Sleeping
Update image_gen.py
Browse files- image_gen.py +2 -1
image_gen.py
CHANGED
|
@@ -38,7 +38,8 @@ import seaborn as sns
|
|
| 38 |
logging.basicConfig(level=logging.INFO)
|
| 39 |
logger = logging.getLogger(__name__)
|
| 40 |
|
| 41 |
-
|
|
|
|
| 42 |
|
| 43 |
def is_valid_png(file_path):
|
| 44 |
"""Check if the PNG file at `file_path` is valid."""
|
|
|
|
| 38 |
logging.basicConfig(level=logging.INFO)
|
| 39 |
logger = logging.getLogger(__name__)
|
| 40 |
|
| 41 |
+
token = os.getenv('HF_API')
|
| 42 |
+
headers = {"Authorization": f"Bearer {token}"}
|
| 43 |
|
| 44 |
def is_valid_png(file_path):
|
| 45 |
"""Check if the PNG file at `file_path` is valid."""
|