Pulastya0 commited on
Commit
e2b7d41
·
1 Parent(s): 1f90ee9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -31,7 +31,6 @@ root_path = os.getenv("HF_SPACE_ROOT_PATH", "")
31
  # --------------------------------------------------------------------
32
  class UserProfile(BaseModel):
33
  skills: List[str] = Field(..., example=["python", "data analysis"])
34
- sectors: List[str] = Field(..., example=["machine learning", "web development"])
35
  internshipType: str = Field(..., example="Work from Home")
36
 
37
  class SearchQuery(BaseModel):
@@ -182,7 +181,7 @@ def get_profile_recommendations(profile: UserProfile):
182
  if chroma_collection is None or encoder is None:
183
  raise HTTPException(status_code=503, detail="Server is not ready.")
184
 
185
- query_text = f"Skills: {', '.join(profile.skills)}. Sectors: {', '.join(profile.sectors)}. Internship Type: {profile.internshipType}"
186
  query_embedding = encoder.encode([query_text])[0].tolist()
187
 
188
  results = chroma_collection.query(
 
31
  # --------------------------------------------------------------------
32
  class UserProfile(BaseModel):
33
  skills: List[str] = Field(..., example=["python", "data analysis"])
 
34
  internshipType: str = Field(..., example="Work from Home")
35
 
36
  class SearchQuery(BaseModel):
 
181
  if chroma_collection is None or encoder is None:
182
  raise HTTPException(status_code=503, detail="Server is not ready.")
183
 
184
+ query_text = f"Skills: {', '.join(profile.skills)}. Internship Type: {profile.internshipType}"
185
  query_embedding = encoder.encode([query_text])[0].tolist()
186
 
187
  results = chroma_collection.query(