Update app.py

app.py

@@ -1,13 +1,7 @@
 import gradio as gr
 
-def classify_job_description(text):
-    import transformers
-    from transformers import pipeline
-
-    nlp = pipeline("text-classification", model="your_model_name")
-    result = nlp(text)
-    # Process the result to extract labels and confidences
-    return result
+# Load the model
+model = gr.load("models/AhmedBou/JoBert")
 
 # Define the title and text labels
 title = "Classify Job Descriptions at the Span Level"
@@ -21,6 +15,9 @@ label_texts = {
 }
 
 # Create the Gradio interface
+def classify_job_description(text):
+    return model(text)
+
 interface = gr.Interface(
     fn=classify_job_description,
     inputs=gr.components.Textbox(lines=10, placeholder="Enter job description here..."),