Update app.py

app.py

@@ -63,12 +63,11 @@ if user_type == 'New User':
         recommendations = recommend_laptops(age=new_user_age, category=new_user_category, gender=new_user_gender)
         st.subheader("Top 5 Recommended Laptops:")
         # for i, row in recommendations.iterrows():
-        recommendations_table = recommendations[['Laptop_Name', 'Price (in Indian Rupees)', 'Type', 'Dedicated Graphic Memory Capacity',
+        recommendations_table = recommendations[['name', 'Price (in Indian Rupees)', 'Type', 'Dedicated Graphic Memory Capacity',
                                       'Processor Brand', 'SSD', 'RAM (in GB)', 'RAM Type', 'Expandable Memory',
                                       'Operating System', 'Touchscreen', 'Screen Size (in inch)', 'Weight (in kg)',
                                       'Refresh Rate', 'screen_resolution', 'company', 'Storage', 'Processor name',
-                                      'CPU_ranking', 'battery_backup', 'gpu name ', 'gpu_benchmark',
-                                      'ram_type_tokenized', 'gpu_processor_tokenized', 'link']]
+                                      'CPU_ranking', 'battery_backup', 'gpu name ', 'gpu_benchmark', 'link']]
 
         st.table(recommendations_table)
 
@@ -83,12 +82,11 @@ elif user_type == 'Existing User':
             recommendations = recommend_laptops(user_id=int(existing_user_id))
             st.subheader(f"Top 5 Recommended Laptops for User {existing_user_id}:")
             # for i, row in recommendations.iterrows():
-            recommendations_table = recommendations[['Laptop_Name', 'Price (in Indian Rupees)', 'Type', 'Dedicated Graphic Memory Capacity',
+            recommendations_table = recommendations[['name', 'Price (in Indian Rupees)', 'Type', 'Dedicated Graphic Memory Capacity',
                                       'Processor Brand', 'SSD', 'RAM (in GB)', 'RAM Type', 'Expandable Memory',
                                       'Operating System', 'Touchscreen', 'Screen Size (in inch)', 'Weight (in kg)',
                                       'Refresh Rate', 'screen_resolution', 'company', 'Storage', 'Processor name',
-                                      'CPU_ranking', 'battery_backup', 'gpu name ', 'gpu_benchmark',
-                                      'ram_type_tokenized', 'gpu_processor_tokenized', 'link']]
+                                      'CPU_ranking', 'battery_backup', 'gpu name ', 'gpu_benchmark', 'link']]
 
             st.table(recommendations_table)
         else: