Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,8 @@ model.config.pad_token_id = model.config.eos_token_id
|
|
11 |
|
12 |
system_prompt = f"<|im_start|>system\nYou are Santa.<|im_end|>\n"
|
13 |
|
|
|
|
|
14 |
@spaces.GPU(duration=120)
|
15 |
def chat(prompt):
|
16 |
input_text = history + "<|im_start|>user\n" + prompt + "<|im_end|>\n"
|
@@ -37,5 +39,4 @@ demo = gr.Interface(
|
|
37 |
)
|
38 |
|
39 |
if __name__ == "__main__":
|
40 |
-
history = system_prompt
|
41 |
demo.launch()
|
|
|
11 |
|
12 |
system_prompt = f"<|im_start|>system\nYou are Santa.<|im_end|>\n"
|
13 |
|
14 |
+
history = system_prompt
|
15 |
+
|
16 |
@spaces.GPU(duration=120)
|
17 |
def chat(prompt):
|
18 |
input_text = history + "<|im_start|>user\n" + prompt + "<|im_end|>\n"
|
|
|
39 |
)
|
40 |
|
41 |
if __name__ == "__main__":
|
|
|
42 |
demo.launch()
|