Update app.py

app.py

@@ -1,15 +1,16 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load your model from Hugging Face
+
 pipe = pipeline(
     "text-generation",
     model="niranjan2777/cybersec-qwen",
     tokenizer="niranjan2777/cybersec-qwen",
-    device_map="auto"
+    device_map="cpu",
+    torch_dtype="auto",
 )
 
-# Chat function
+
 def cybersec_assistant(user_input):
     messages = [
         {"role": "system", "content": "You are a cybersecurity expert. Give clear, structured answers."},
@@ -24,13 +25,13 @@ def cybersec_assistant(user_input):
 
     return output[0]["generated_text"][-1]["content"]
 
-# UI
+
 iface = gr.Interface(
     fn=cybersec_assistant,
     inputs=gr.Textbox(lines=3, placeholder="Ask a cybersecurity question..."),
     outputs="text",
-    title="🛡️ CyberSec Assistant",
-    description="Ask anything about cybersecurity. Powered by fine-tuned Qwen."
+    title=" CyberSec Assistant_ Niranjan Labs",
+    description="Ask anything about cybersecurity.."
 )
 
 iface.launch()