Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
 from transformers import pipeline
+import torch  # <-- ADD THIS LINE
 
 # Load GPT-2 with optimized parameters
 model = pipeline(
     "text-generation",
     model="gpt2",
-    max_length=200,          # Maximum response length
-    temperature=0.7,         # Creativity (0.0 = deterministic, 1.0 = random)
-    early_stopping=True,     # Stop generation early if possible
+    max_length=200,
+    temperature=0.7,
+    early_stopping=True,
     torch_dtype=torch.float32  # CPU compatibility
 )
 
 def chat_with_gpt2(user_input):
-    # Generate a response based on user input
     prompt = f"{user_input}"
     try:
         with torch.no_grad():