Update app.py

app.py

@@ -72,7 +72,8 @@ def pdf2text(file_path):
   return clean_text(text)
 
 
-def ner(text) :
+def ner(text, labels, threshold) :
+    labels = labels.split(",")
     return {
         "text": text,
         "entities": [
@@ -84,14 +85,14 @@ def ner(text) :
                 "score": 0,
             }
             for entity in model.predict_entities(
-                text, labels, flat_ner=False, threshold=0.27
+                text, labels, flat_ner=False, threshold=threshold
             )
         ],
     }
 
-def parser(file_path):
+def parser(file_path, labels):
   text = pdf2text(file_path)
-  return ner(text)
+  return ner(text, labels, threshold)
 
 
 # Define a custom CSS style
@@ -121,7 +122,25 @@ h1 {
 with gr.Blocks(css=custom_css) as demo:
     gr.HTML("<h1>AI-Powered Resume Parser</h1>")
     gr.HTML("<p style='text-align: center;'>This tool uses advanced NLP techniques to extract key information from your resume.</p>")
+
     
+    with gr.Row() as row:
+        labels = gr.Textbox(
+            value=examples[0][1],
+            label="Labels",
+            placeholder="Enter your labels here (comma separated)",
+            scale=1,
+        )
+        threshold = gr.Slider(
+            0,
+            1,
+            value=0.3,
+            step=0.01,
+            label="Threshold",
+            info="Lower the threshold to increase how many entities get predicted.",
+            scale=0,
+        )
+
     with gr.Row():
         file_input = gr.File(label="Upload Resume",
                 file_types=['.pdf'],
@@ -136,7 +155,7 @@ with gr.Blocks(css=custom_css) as demo:
                            combine_adjacent=True
                            )
 
-    parse_button.click(fn=parser, inputs=file_input, outputs=output)
+    parse_button.click(fn=parser, inputs=[file_input,labels, threshold], outputs=output)
 
     gr.HTML("<p style='text-align: center;'>Our resume parser can identify and extract important details such as personal information, education, work experience, skills, and more. Simply upload your resume and let our AI do the work!</p>")