Update app.py

app.py

@@ -1,15 +1,14 @@
 import gradio as gr
+import torch  # ✅ Explicitly import torch
 from transformers import pipeline
 import re
 
-# Load open-source LLM
+# Load model
 llm = pipeline("text-generation", model="gpt2")
 
-# Define guardrails
+# Guardrails
 blocked_keywords = ["kill", "bomb", "hack", "suicide", "drugs"]
 blocked_patterns = [r"\bhow to (make|build|create) (a )?(bomb|weapon)\b", r"\b(hack|bypass) (password|system)\b"]
-
-# Violation log
 violation_log = []
 
 def guardrails_check(prompt):