sotosbarl commited on
Commit
67dbaea
·
verified ·
1 Parent(s): 26c7407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ def init_session_state():
13
  # Initialize session state
14
  init_session_state()
15
 
16
- pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 
17
  # pipe = pipeline("text-generation", model="GeneZC/MiniChat-1.5-3B")
18
  # pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
19
  # model_name = "MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
 
13
  # Initialize session state
14
  init_session_state()
15
 
16
+ # pipe = pipeline("text2text-generation", model="google/flan-t5-base")
17
+ pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
18
  # pipe = pipeline("text-generation", model="GeneZC/MiniChat-1.5-3B")
19
  # pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
20
  # model_name = "MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"