sotosbarl commited on
Commit
3356d81
1 Parent(s): d06aa9f

Update app.py

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