Update app.py

app.py

@@ -1,27 +1,39 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-# Load the summarization pipeline from Hugging Face
-rephrase_pipeline = pipeline("text2text-generation", model="Vamsi/T5_Paraphrase_Paws")
+# Load model and tokenizer
+@st.cache_resource
+def load_pipeline():
+    model_name = "Vamsi/T5_Paraphrase_Paws"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    return pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+
+rephrase_pipeline = load_pipeline()
 
 def tailor_resume(job_description, resume):
-    prompt = f"Job Description: {job_description}\nResume: {resume}\n\nRewrite the resume to match the job description:"
-    output = rephrase_pipeline(prompt, max_length=512, do_sample=True, top_k=50)[0]["generated_text"]
+    # Create a concise prompt
+    prompt = f"paraphrase: Adapt this resume for the job description.\nJob: {job_description}\nResume: {resume}"
+    output = rephrase_pipeline(prompt, max_length=512, num_return_sequences=1, clean_up_tokenization_spaces=True)[0]["generated_text"]
     return output
 
+# Streamlit UI
 st.set_page_config(page_title="Resume Tailor", layout="centered")
 
 st.title("🎯 Resume Tailoring App")
-st.write("Enhance your resume based on the job description using AI.")
+st.write("Use AI to align your resume with a specific job description.")
 
-job_desc = st.text_area("📋 Paste the Job Description here:", height=200)
-resume_input = st.text_area("🧾 Paste Your Resume here:", height=300)
+job_desc = st.text_area("📋 Paste the Job Description:", height=200)
+resume_input = st.text_area("🧾 Paste Your Resume:", height=300)
 
 if st.button("Tailor Resume"):
-    if job_desc.strip() == "" or resume_input.strip() == "":
+    if not job_desc.strip() or not resume_input.strip():
         st.warning("Please enter both the job description and your resume.")
     else:
         with st.spinner("Tailoring your resume..."):
-            tailored_resume = tailor_resume(job_desc, resume_input)
-            st.subheader("🎉 Tailored Resume")
-            st.text_area("You can now copy your optimized resume below:", value=tailored_resume, height=400)
+            try:
+                tailored_resume = tailor_resume(job_desc, resume_input)
+                st.subheader("🎉 Tailored Resume")
+                st.text_area("You can now copy your optimized resume:", value=tailored_resume, height=400)
+            except Exception as e:
+                st.error(f"Something went wrong: {e}")