Update main.py

main.py

@@ -32,7 +32,7 @@ root_path = os.getenv("HF_SPACE_ROOT_PATH", "")
 class UserProfile(BaseModel):
     skills: List[str] = Field(..., example=["python", "data analysis"])
     sectors: List[str] = Field(..., example=["machine learning", "web development"])
-    internshipType: str = Field(..., example="Bengaluru")
+    internshipType: str = Field(..., example="Work from Home")
 
 class SearchQuery(BaseModel):
     query: str = Field(..., example="marketing internship in mumbai")
@@ -182,7 +182,7 @@ def get_profile_recommendations(profile: UserProfile):
     if chroma_collection is None or encoder is None:
         raise HTTPException(status_code=503, detail="Server is not ready.")
 
-    query_text = f"Skills: {', '.join(profile.skills)}. Sectors: {', '.join(profile.sectors)}. internshipType: {profile.location}"
+    query_text = f"Skills: {', '.join(profile.skills)}. Sectors: {', '.join(profile.sectors)}. Internship Type: {profile.internshipType}"
     query_embedding = encoder.encode([query_text])[0].tolist()
     
     results = chroma_collection.query(