Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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)}.
|
| 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(
|