awinml commited on
Commit
b241a5c
1 Parent(s): 26de0f1

Upload utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -36,7 +36,7 @@ def get_spacy_model():
36
 
37
  @st.experimental_singleton
38
  def get_flan_alpaca_xl_model():
39
- return pipeline('text2text-generation', model="./models/flan-alpaca-xl")
40
 
41
 
42
  # Initialize models from HuggingFace
 
36
 
37
  @st.experimental_singleton
38
  def get_flan_alpaca_xl_model():
39
+ return pipeline(model="./models/flan-alpaca-xl")
40
 
41
 
42
  # Initialize models from HuggingFace