Update app.py

app.py

@@ -1,8 +1,17 @@
 import gradio as gr
+from fake_job_detector.models import DistilBERTBaseModel
+
 
 def process_input(title, description):
-    """Concatenates the title and description."""
-    return f"Title: {title}\nDescription: {description}"
+# Load the safetensors
+    safetensors_path = "base_unfrozen_5epoch"
+
+# Load the DistilBERT model
+    model = DistilBERTBaseModel(safetensors_path)
+
+    result = model(title, description)
+    
+    return result
 
 def main():
     # Define the interface