nroggendorff commited on
Commit
d1d7004
·
verified ·
1 Parent(s): 1e38c28

Update app.py

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