andyfe commited on
Commit
7ace03c
·
1 Parent(s): 5d80cfa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -158,7 +158,9 @@ def query(system_prompt, user_prompt, temperature=0.9, max_new_tokens=256, top_p
158
  )
159
  prompt = f"System: {system_prompt}\nUser: {user_prompt}\n"
160
  print(prompt)
 
161
  output = client.text_generation(prompt, **generate_kwargs, details=True, return_full_text=False)
 
162
  # stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
163
  # output = ""
164
 
 
158
  )
159
  prompt = f"System: {system_prompt}\nUser: {user_prompt}\n"
160
  print(prompt)
161
+ print('-----')
162
  output = client.text_generation(prompt, **generate_kwargs, details=True, return_full_text=False)
163
+ print(output)
164
  # stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
165
  # output = ""
166