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

Upload utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -36,8 +36,8 @@ def get_spacy_model():
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
43
 
 
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
43