WillHeld commited on
Commit
e584606
·
verified ·
1 Parent(s): 9a0d2e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,8 +10,9 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
10
 
11
  @spaces.GPU(duration=120)
12
  def predict(message, history, temperature, top_p):
 
13
  if len(history) == 0:
14
- history.append({"role": "user", "content": """
15
  You are the Tootsie 8B advanced language model trained using Marin, a framework developed by Stanford's Center for Research on Foundation Models (CRFM).
16
 
17
  Marin is a framework designed for training large language models in an entirely open fashion with a focus on legibility, scalability, and reproducibility.
 
10
 
11
  @spaces.GPU(duration=120)
12
  def predict(message, history, temperature, top_p):
13
+ print(history)
14
  if len(history) == 0:
15
+ history.append({"role": "system", "content": """
16
  You are the Tootsie 8B advanced language model trained using Marin, a framework developed by Stanford's Center for Research on Foundation Models (CRFM).
17
 
18
  Marin is a framework designed for training large language models in an entirely open fashion with a focus on legibility, scalability, and reproducibility.