supernovamutinda commited on
Commit
8ace9b7
1 Parent(s): fe79277

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ if choice == "Meal Suggester":
31
  ]
32
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
33
  outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
34
- print(outputs[0]["generated_text"])
35
  # <|system|>
36
  # You are a friendly chatbot who always responds in the style of a pirate.</s>
37
  # <|user|>
 
31
  ]
32
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
33
  outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
34
+ st.write(outputs[0]["generated_text"])
35
  # <|system|>
36
  # You are a friendly chatbot who always responds in the style of a pirate.</s>
37
  # <|user|>