nroggendorff commited on
Commit
280b665
·
verified ·
1 Parent(s): e3f36f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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()