File size: 2,178 Bytes
1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 22cfd3c 1dae8e9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import pandas as pd
import gradio as gr
import matplotlib.pyplot as plt
import io
import base64
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import linear_kernel
data = pd.read_csv('Blinkit Cart Prediction.csv')
tfidf_vectorizer = TfidfVectorizer(max_features=1000) # Adjust max_features as needed
tfidf_matrix = tfidf_vectorizer.fit_transform(data['Description'])
def recommend_products(user_choice, num_recommendations=10):
user_choice_vector = tfidf_vectorizer.transform([user_choice])
cosine_similarities = linear_kernel(user_choice_vector, tfidf_matrix)
similar_indices = cosine_similarities.argsort()[0][-num_recommendations - 1:-1][::-1]
recommended_products = data.iloc[similar_indices][['ProductID', 'ProductName']]
return recommended_products
input_component = gr.inputs.Textbox(label="Enter Your Choice")
output_component = gr.outputs.HTML(label="Recommended Products")
def recommend_interface(user_choice):
recommended_products = recommend_products(user_choice)
plt.figure(figsize=(10, 6))
plt.bar(recommended_products['ProductName'], range(len(recommended_products)), color='skyblue')
plt.xticks(rotation=45, ha="right")
plt.xlabel("Recommended Products")
plt.ylabel("Ranking")
plt.title("Top Recommended Products")
buffer = io.BytesIO()
plt.savefig(buffer, format="png")
graph_base64 = base64.b64encode(buffer.getvalue()).decode()
plt.close()
graph_html = f'<img src="data:image/png;base64,{graph_base64}" />'
table_html = recommended_products.to_html(index=False)
result_html = f"<h2>Recommended Products:</h2>{table_html}<br>{graph_html}"
return result_html
interface = gr.Interface(
fn=recommend_interface, inputs=input_component, outputs=output_component,
live = True,
description = " Press flag if any erroneous output comes ",
theme=gr.themes.Soft(),
title = "Blinkit Cart Prediction",
examples = [['necklace'],
['DSLR camera '],['tea '], ['Smart TV '] , ['protein bars'] , ['sunglasses '] ],
)
interface.launch(inline=False) |