Shritam commited on
Commit
fdbd33b
·
1 Parent(s): 125b07e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,9 +2,11 @@ import gradio as gr
2
  import pandas as pd
3
  import numpy as np
4
  import pickle
5
- #from sentence_transformers import SentenceTransformer
6
  from sklearn.metrics.pairwise import cosine_similarity
7
 
 
 
8
  # Load the dataset
9
  data = pd.read_csv('data/Recommender_data.csv')
10
 
 
2
  import pandas as pd
3
  import numpy as np
4
  import pickle
5
+ from sentence_transformers import SentenceTransformer
6
  from sklearn.metrics.pairwise import cosine_similarity
7
 
8
+ model = SentenceTransformer('paraphrase-distilroberta-base-v1')
9
+
10
  # Load the dataset
11
  data = pd.read_csv('data/Recommender_data.csv')
12