Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,13 @@ from langchain_openai import ChatOpenAI
|
|
16 |
from typing import List, Dict, Optional
|
17 |
from pydantic import BaseModel, Field, confloat
|
18 |
|
19 |
-
llm = LLM(model="deepseek/deepseek-r1-distill-qwen-32b")
|
|
|
20 |
|
|
|
|
|
|
|
|
|
21 |
class SkillScore(BaseModel):
|
22 |
skill_name: str = Field(description="Name of the skill being scored")
|
23 |
required: bool = Field(description="Whether this skill is required or nice-to-have")
|
|
|
16 |
from typing import List, Dict, Optional
|
17 |
from pydantic import BaseModel, Field, confloat
|
18 |
|
19 |
+
# llm = LLM(model="deepseek/deepseek-r1-distill-qwen-32b")
|
20 |
+
from crewai.llms import LiteLLM
|
21 |
|
22 |
+
llm = LiteLLM(
|
23 |
+
model="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
|
24 |
+
api_base="http://localhost:8000" # or DeepSeek API
|
25 |
+
)
|
26 |
class SkillScore(BaseModel):
|
27 |
skill_name: str = Field(description="Name of the skill being scored")
|
28 |
required: bool = Field(description="Whether this skill is required or nice-to-have")
|