ZennethKenneth commited on
Commit
4663596
·
verified ·
1 Parent(s): 1482961

update kwargs

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -10,9 +10,8 @@ if not hf_token:
10
 
11
  def authenticate_and_generate(message, history, system_message, max_tokens, temperature, top_p):
12
  try:
13
- # Initialize the text-generation pipeline with the provided token
14
- # Test with a known public model
15
- text_generator = pipeline("text-generation", model="gpt2", use_auth_token=hf_token)
16
 
17
  # Ensure that system_message is a string
18
  system_message = str(system_message)
 
10
 
11
  def authenticate_and_generate(message, history, system_message, max_tokens, temperature, top_p):
12
  try:
13
+ # Initialize the text-generation pipeline without use_auth_token if not needed
14
+ text_generator = pipeline("text-generation", model="gpt2")
 
15
 
16
  # Ensure that system_message is a string
17
  system_message = str(system_message)