Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ krishnamurti_generator = pipeline("text-generation", model="distilgpt2")
|
|
18 |
############### DEFINING FUNCTIONS ###########################
|
19 |
|
20 |
def ask_krishnamurti(question):
|
21 |
-
answer = krishnamurti_generator(question, min_length=
|
22 |
-
answer = " ".join(sent_tokenize(answer)[:
|
23 |
return answer
|
24 |
|
25 |
def get_similar_quotes(question):
|
|
|
18 |
############### DEFINING FUNCTIONS ###########################
|
19 |
|
20 |
def ask_krishnamurti(question):
|
21 |
+
answer = krishnamurti_generator(question, min_length=100, max_length=120)[0]['generated_text'] # generate about 50 word tokens
|
22 |
+
answer = " ".join(sent_tokenize(answer)[:6]) # Get the first five sentences
|
23 |
return answer
|
24 |
|
25 |
def get_similar_quotes(question):
|