Update app.py

app.py

@@ -1,3 +1,30 @@
 import gradio as gr
 
-gr.load("models/AhmedBou/JoBert").launch()
+# Load the model
+model = gr.load("models/AhmedBou/JoBert")
+
+# Define the title and text labels
+title = "Classify Job Descriptions at the Span Level"
+label_texts = {
+    "Label_0": "About the Company:",
+    "Label_1": "Job Description:",
+    "Label_2": "Job Requirements:",
+    "Label_3": "Responsibilities:",
+    "Label_4": "Benefits:",
+    "Label_5": "Other:"
+}
+
+# Create the Gradio interface
+def classify_job_description(text):
+    return model(text)
+
+interface = gr.Interface(
+    fn=classify_job_description,
+    inputs=gr.inputs.Textbox(lines=10, placeholder="Enter job description here..."),
+    outputs=gr.outputs.Textbox(label="Classification Results"),
+    title=title,
+    description="\n".join([f"{key}: {value}" for key, value in label_texts.items()])
+)
+
+# Launch the interface
+interface.launch()