Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def format_prompt(base, user_message):
|
|
31 |
|
32 |
|
33 |
def predict(input, history=[]):
|
34 |
-
input =
|
35 |
new_user_input_ids = tokenizer.encode(f"{input}{tokenizer.eos_token}", return_tensors="pt")
|
36 |
bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
|
37 |
|
|
|
31 |
|
32 |
|
33 |
def predict(input, history=[]):
|
34 |
+
input = format_prompt(get_base_prompt(), input)
|
35 |
new_user_input_ids = tokenizer.encode(f"{input}{tokenizer.eos_token}", return_tensors="pt")
|
36 |
bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
|
37 |
|