Update app.py

app.py

@@ -5,8 +5,31 @@ import openai
 openai.api_key = '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'  # Remember never to expose API keys in code
 
 KNOWN_MODELS = [
+    # General ML models
     "Neural Networks", "Decision Trees", "Support Vector Machines", 
-    "Random Forests", "Linear Regression", "Reinforcement Learning"
+    "Random Forests", "Linear Regression", "Reinforcement Learning",
+    "Logistic Regression", "k-Nearest Neighbors", "Naive Bayes",
+    "Gradient Boosting Machines", "Regularization Techniques", 
+    "Ensemble Methods", "Time Series Analysis",
+
+    # Deep Learning models
+    "Deep Learning", "Convolutional Neural Networks", 
+    "Recurrent Neural Networks", "Transformer Models", 
+    "Generative Adversarial Networks", "Autoencoders", 
+    "Bidirectional LSTM", "Residual Networks (ResNets)", 
+    "Variational Autoencoders",
+
+    # Computer Vision models and techniques
+    "Object Detection (e.g., YOLO, SSD)", "Semantic Segmentation",
+    "Image Classification", "Face Recognition", "Optical Character Recognition (OCR)", 
+    "Pose Estimation", "Style Transfer", "Image-to-Image Translation", 
+    "Image Generation", "Capsule Networks",
+
+    # NLP models and techniques
+    "BERT", "GPT", "ELMo", "T5", "Word2Vec", "Doc2Vec", 
+    "Topic Modeling", "Sentiment Analysis", "Text Classification", 
+    "Machine Translation", "Speech Recognition", "Sequence-to-Sequence Models", 
+    "Attention Mechanisms", "Named Entity Recognition", "Text Summarization"
 ]
 
 def recommend_ai_model_via_gpt(description):