sotosbarl commited on
Commit
d06aa9f
1 Parent(s): eec15b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ def init_session_state():
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)
20
  # model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
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)
20
  # model = AutoModelForSequenceClassification.from_pretrained(model_name)