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)
|