eliwill commited on
Commit
98f174a
·
1 Parent(s): 0305cba

Add Krishnamurti Model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = SentenceTransformer(sentence_similarity_model)
17
 
18
  # Loading in text-generation models
19
  stoic_generator = pipeline("text-generation", model="eliwill/stoic-generator-10e")
20
- krishnamurti_generator = pipeline("text-generation", model="distilgpt2")
21
 
22
  # Creating philosopher dictionary
23
  philosopher_dictionary = {
 
17
 
18
  # Loading in text-generation models
19
  stoic_generator = pipeline("text-generation", model="eliwill/stoic-generator-10e")
20
+ krishnamurti_generator = pipeline("text-generation", model="eliwill/distilgpt2-finetuned-final-project")
21
 
22
  # Creating philosopher dictionary
23
  philosopher_dictionary = {