Bofandra commited on
Commit
25fc91b
·
verified ·
1 Parent(s): ef8f23c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,12 +1,10 @@
1
  import gradio as gr
2
  from text_generation import InferenceAPIClient
3
  import os
 
4
 
5
  # ✅ Choose your model (you can change this to another instruct model)
6
- client = InferenceAPIClient(
7
- "deepseek-ai/DeepSeek-R1",
8
- api_key=os.getenv("HF_TOKEN")
9
- )
10
 
11
  # 🧠 Function to generate software architecture
12
  def generate_software_spec(name, description, architecture, components, deployment, platform, extra):
 
1
  import gradio as gr
2
  from text_generation import InferenceAPIClient
3
  import os
4
+ os.environ["HF_TOKEN"] = "hf_your_token_here"
5
 
6
  # ✅ Choose your model (you can change this to another instruct model)
7
+ client = InferenceAPIClient("deepseek-ai/DeepSeek-R1")
 
 
 
8
 
9
  # 🧠 Function to generate software architecture
10
  def generate_software_spec(name, description, architecture, components, deployment, platform, extra):