Update app.py

app.py

@@ -1,17 +1,23 @@
 from gradio import components as gc
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-# Create a sidebar component
-sidebar = gc.SideBar()
-for item in sidebar_items:
-    prompt_button = gc.Button(text=item["prompt"])
-    description = item["description"]
-    # Add additional styling or functionality if needed
+# Load model and tokenizer
+model_name = "Canstralian/CySec_Known_Exploit_Analyzer"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+# Define the function for text input processing
+def greet(text):
+    # Tokenize and process the input text
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model(**inputs)
     
-    # Append the button and description to the sidebar component
-    sidebar.append(prompt_button, description)
+    # Extract the label with the highest score
+    predicted_label = outputs.logits.argmax().item()
+    return f"Greeting, {text}! Predicted label: {predicted_label}"
 
-# Create the interface without a sidebar (since we've added it separately)
+# Create the interface
 iface = gr.Interface(
     fn=greet,
     inputs="text",
@@ -20,7 +26,9 @@ iface = gr.Interface(
     description="Ask a user for their name and greet them."
 )
 
-# Add the left sidebar to the interface
+# Optional: define and add a sidebar if needed
+# Example sidebar component (replace with your intended content)
+sidebar = gr.Textbox(label="Sidebar Info")
 iface.add_component(sidebar, side="left")
 
 # Launch the Gradio app