Update ResumeOpt.py

ResumeOpt.py

@@ -14,7 +14,8 @@ except IOError:
 #nlp = spacy.load('en_core_web_sm')
 # Load models quietly
 #nlp = spacy.load('en_core_web_sm')
-fill_mask = pipeline("fill-mask", model="bert-base-uncased", device=0)  # Use GPU if available
+device = 0 if torch.cuda.is_available() else -1
+fill_mask = pipeline("fill-mask", model="bert-base-uncased", device=device)  # Use GPU if available
 
 # Cell 3: Define functions
 def extract_text_from_docx(file_path):