Update app.py

app.py

@@ -9,7 +9,7 @@ try:
 except ImportError:
     raise RuntimeError("Missing required dependency: flash_attn. Install with `pip install flash-attn --no-build-isolation`")
 
-# Load DeepSeek-R1 model with trust_remote_code enabled
+# Load DeepSeek-R1 model
 model_name = "deepseek-ai/DeepSeek-R1"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
@@ -26,7 +26,7 @@ model = AutoModelForCausalLM.from_pretrained(
 # Use a text-generation pipeline
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
 
-# 🎯 Function to extract interests from user input
+# 🎯 Extract interests from user input
 def extract_interests(text):
     prompt = f"Extract 3-5 relevant interests from this request: '{text}'. Focus on hobbies and product preferences."
 
@@ -76,3 +76,5 @@ demo = gr.Interface(
 # 🚀 Launch Gradio App
 if __name__ == "__main__":
     demo.launch()
+
+