raannakasturi commited on
Commit
6b83232
·
1 Parent(s): eaa1129

Load environment variables and use HF_TOKEN for Hugging Face API authentication in generate_svg.py

Browse files
Files changed (1) hide show
  1. 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}",