Update app.py

app.py

@@ -1,11 +1,22 @@
 import streamlit as st
-import re, torch,os,ast
-from datetime import datetime
+import re, ast
+import pandas as pd
 from huggingface_hub import login, InferenceClient
-import random
+import pickle
+from sklearn.metrics.pairwise import cosine_similarity
+
+st.set_page_config(layout="wide")
 
 login(token=os.getenv("TOKEN"))
 
+with open('l.pkl', 'rb') as file:
+    similarity = pickle.load(file)
+
+with open('items_dict.pkl', 'rb') as file:
+    items_dict = pickle.load(file)
+
+data=pd.read_csv('marketing_sample_for_walmart_com-product_details__20200101_20200331__30k_data.csv')
+
 # Initialize the inference client for the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")