Files changed (1) hide show
  1. utils.py +6 -0
utils.py CHANGED
@@ -35,6 +35,12 @@ def get_flan_t5_model():
35
  "summarization", model="google/flan-t5-small", tokenizer="google/flan-t5-small"
36
  )
37
 
 
 
 
 
 
 
38
 
39
  @st.experimental_singleton
40
  def get_mpnet_embedding_model():
 
35
  "summarization", model="google/flan-t5-small", tokenizer="google/flan-t5-small"
36
  )
37
 
38
+ def get_gptj():
39
+ return pipeline(
40
+ "summarization", model="EleutherAI/gpt-j-6B", tokenizer="EleutherAI/gpt-j-6B"
41
+ )
42
+
43
+
44
 
45
  @st.experimental_singleton
46
  def get_mpnet_embedding_model():