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}",
|