Fix quote length
Browse files
app.py
CHANGED
@@ -10,12 +10,12 @@ df = pd.read_json("krishnamurti_df.json")
|
|
10 |
# Loading back in our sentence similarity and language model
|
11 |
model = SentenceTransformer("msmarco-roberta-base-v3") # best performing model
|
12 |
|
13 |
-
krishnamurti_generator = pipeline("text-generation", model="distilgpt2")
|
14 |
|
15 |
############### DEFINING FUNCTIONS ###########################
|
16 |
|
17 |
def ask_krishnamurti(question):
|
18 |
-
answer = krishnamurti_generator(question)[0]['generated_text']
|
19 |
return answer
|
20 |
|
21 |
def get_similar_quotes(question):
|
@@ -24,6 +24,7 @@ def get_similar_quotes(question):
|
|
24 |
ind = np.argpartition(sims, -5)[-5:]
|
25 |
similar_sentences = [df['Quotes'][i] for i in ind]
|
26 |
top5quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,6))
|
|
|
27 |
return top5quotes
|
28 |
|
29 |
def main(question):
|
|
|
10 |
# Loading back in our sentence similarity and language model
|
11 |
model = SentenceTransformer("msmarco-roberta-base-v3") # best performing model
|
12 |
|
13 |
+
krishnamurti_generator = pipeline("text-generation", model="distilgpt2",)
|
14 |
|
15 |
############### DEFINING FUNCTIONS ###########################
|
16 |
|
17 |
def ask_krishnamurti(question):
|
18 |
+
answer = krishnamurti_generator(question,)[0]['generated_text']
|
19 |
return answer
|
20 |
|
21 |
def get_similar_quotes(question):
|
|
|
24 |
ind = np.argpartition(sims, -5)[-5:]
|
25 |
similar_sentences = [df['Quotes'][i] for i in ind]
|
26 |
top5quotes = pd.DataFrame(data = similar_sentences, columns=["Quotes"], index=range(1,6))
|
27 |
+
top5quotes['Quotes'] = top5quotes['Quotes'].str[:500] + "..."
|
28 |
return top5quotes
|
29 |
|
30 |
def main(question):
|