Spaces:
Sleeping
Sleeping
Commit ·
6b83232
1
Parent(s): eaa1129
Load environment variables and use HF_TOKEN for Hugging Face API authentication in generate_svg.py
Browse files- generate_svg.py +6 -1
generate_svg.py
CHANGED
|
@@ -2,8 +2,13 @@ import os
|
|
| 2 |
from graphviz import Digraph
|
| 3 |
# from cairosvg import svg2pdf
|
| 4 |
import re
|
|
|
|
|
|
|
| 5 |
import random
|
| 6 |
|
|
|
|
|
|
|
|
|
|
| 7 |
def parse_markdown_to_dict(md_text):
|
| 8 |
lines = md_text.strip().splitlines()
|
| 9 |
mindmap = {}
|
|
@@ -119,7 +124,7 @@ def generate_mindmap(md_text):
|
|
| 119 |
|
| 120 |
def upload_svg(mindmap_svg):
|
| 121 |
from huggingface_hub import HfApi
|
| 122 |
-
api = HfApi()
|
| 123 |
api.upload_file(
|
| 124 |
path_or_fileobj=mindmap_svg,
|
| 125 |
path_in_repo=f"SVG/{mindmap_svg}",
|
|
|
|
| 2 |
from graphviz import Digraph
|
| 3 |
# from cairosvg import svg2pdf
|
| 4 |
import re
|
| 5 |
+
import os
|
| 6 |
+
import dotenv
|
| 7 |
import random
|
| 8 |
|
| 9 |
+
dotenv.load_dotenv()
|
| 10 |
+
hf_token = os.getenv("HF_TOKEN")
|
| 11 |
+
|
| 12 |
def parse_markdown_to_dict(md_text):
|
| 13 |
lines = md_text.strip().splitlines()
|
| 14 |
mindmap = {}
|
|
|
|
| 124 |
|
| 125 |
def upload_svg(mindmap_svg):
|
| 126 |
from huggingface_hub import HfApi
|
| 127 |
+
api = HfApi(token=hf_token)
|
| 128 |
api.upload_file(
|
| 129 |
path_or_fileobj=mindmap_svg,
|
| 130 |
path_in_repo=f"SVG/{mindmap_svg}",
|