bainskarman commited on
Commit
ad4e39f
·
verified ·
1 Parent(s): a3e9ce4

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -8
model.py CHANGED
@@ -22,15 +22,8 @@ def modelFeedback(ats_score, resume_data):
22
  input_ids = tokenizer(input_prompt, return_tensors="pt").input_ids
23
  input_length = input_ids.shape[1]
24
 
25
- print(f"Input length: {input_length}")
26
 
27
  # Generate response
28
  response = generator(input_prompt, num_return_sequences=1)
29
-
30
- # Check if response is not empty
31
- if response and len(response) > 0:
32
- generated_text = response[0]['generated_text']
33
- else:
34
- generated_text = "No response generated."
35
- response = generated_text
36
  return response
 
22
  input_ids = tokenizer(input_prompt, return_tensors="pt").input_ids
23
  input_length = input_ids.shape[1]
24
 
 
25
 
26
  # Generate response
27
  response = generator(input_prompt, num_return_sequences=1)
28
+
 
 
 
 
 
 
29
  return response