RoAr777 commited on
Commit
f05f219
·
1 Parent(s): 93b203e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -11,12 +11,8 @@ replacements = {
11
  '<': 'LessThanSign'
12
  }
13
 
14
- from transformers import AutoTokenizer, AutoModelForCausalLM
15
-
16
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
17
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
18
- #model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
19
- #tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
20
 
21
  def chatbot_model(m,history):
22
  for char, replacement in replacements.items():
 
11
  '<': 'LessThanSign'
12
  }
13
 
14
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
15
+ tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
 
 
 
 
16
 
17
  def chatbot_model(m,history):
18
  for char, replacement in replacements.items():