ranamhamoud commited on
Commit
4518db8
·
verified ·
1 Parent(s): 9905ae2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def generate(
63
  for user, assistant in chat_history:
64
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
65
  conversation.append({"role": "user", "content": message})
66
- if model == "A":
67
  model = modelA
68
  tokenizer = tokenizerA
69
  enc = tokenizer(make_prompt(message), return_tensors="pt", padding=True, truncation=True)
 
63
  for user, assistant in chat_history:
64
  conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
65
  conversation.append({"role": "user", "content": message})
66
+ if model == "A":
67
  model = modelA
68
  tokenizer = tokenizerA
69
  enc = tokenizer(make_prompt(message), return_tensors="pt", padding=True, truncation=True)