Update src/streamlit_app.py

src/streamlit_app.py

@@ -80,7 +80,7 @@ def calculate_ats_match(resume_text: str, job_desc_text: str) -> tuple[float, li
     emb_resume = embed(resume_text)
     emb_job = embed(job_desc_text)
 
-    cosine_score = torch.nn.functional.cosine_similarity(emb_resume, emb_job).item()
+    cosine_score = torch.nn.functional.cosine_similarity(emb_resume, emb_job).cpu().item()
     ats_score = round(float(cosine_score) * 100, 2)
     
     job_doc = nlp(job_desc_text.lower())
@@ -142,4 +142,4 @@ if st.button("🔍 Analyze ATS Match Score"):
         else:
             st.success("Your resume covers all high-value concepts. Great job! 🎉")
     else:
-        st.warning("Please paste text into both the Resume and Job Description boxes.")
+        st.warning("Please paste text into both the Resume and Job Description boxes.")