Spaces:
Build error
Build error
Update utils.py
#2
by
varun500
- opened
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():
|