eliwill commited on
Commit
ce1f130
·
1 Parent(s): 0d83ce7

Fix quote length

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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):