Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ system_prompt = f"<|im_start|>system\nYou are Santa.<|im_end|>\n"
|
|
13 |
|
14 |
history = system_prompt
|
15 |
|
16 |
-
@spaces.GPU
|
17 |
def chat(prompt):
|
18 |
input_text = history + "<|im_start|>user\n" + prompt + "<|im_end|>\n"
|
19 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
|
|
13 |
|
14 |
history = system_prompt
|
15 |
|
16 |
+
@spaces.GPU
|
17 |
def chat(prompt):
|
18 |
input_text = history + "<|im_start|>user\n" + prompt + "<|im_end|>\n"
|
19 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|