Added huggingface inference configuration

README.md

@@ -9,6 +9,9 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 short_description: Generate PlantUML diagrams from natural language description
+hf_oauth: true
+hf_oauth_scopes:
+  - inference-api
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -5,15 +5,23 @@ from src.retriever_tool import RetrieverTool
 from src.plantuml import render_plantuml
 from src.examples import get_plantuml_examples
 import re
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+user_token = os.getenv("HF_TOKEN")
 
 PDF_PATH = "PlantUML_Language_Reference_Guide_en.pdf"
 
 docs_processed = get_processed_documents(PDF_PATH)
 retriever_tool = RetrieverTool(docs_processed)
 
+model = InferenceClientModel('microsoft/phi-4', api_key=user_token)
+
 agent = CodeAgent(
     tools=[retriever_tool],
-    model=InferenceClientModel('microsoft/phi-4'),
+    model=model,
     max_steps=4,
     verbosity_level=2,
 )

requirements.txt

@@ -6,4 +6,5 @@ smolagents
 smolagents[gradio]
 plantweb
 rank_bm25
-numpy==1.26.4
+numpy==1.26.4
+python-dotenv