mMonika commited on
Commit
17205d8
·
verified ·
1 Parent(s): 163f146

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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")