Spaces:
Paused
Paused
Update app.py
Browse files
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 |
-
|
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)
|