Update app.py
Browse files
app.py
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from text_generation import InferenceAPIClient
|
|
|
|
| 3 |
|
| 4 |
# ✅ Choose your model (you can change this to another instruct model)
|
| 5 |
-
client = InferenceAPIClient("deepseek-ai/DeepSeek-R1")
|
| 6 |
|
| 7 |
# 🧠 Function to generate software architecture
|
| 8 |
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 |
|
| 5 |
# ✅ Choose your model (you can change this to another instruct model)
|
| 6 |
+
client = InferenceAPIClient("deepseek-ai/DeepSeek-R1", token=os.environ["HF_TOKEN"])
|
| 7 |
|
| 8 |
# 🧠 Function to generate software architecture
|
| 9 |
def generate_software_spec(name, description, architecture, components, deployment, platform, extra):
|