Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|