Spaces:
Running
Running
雷娃
commited on
Commit
·
e08f4f8
1
Parent(s):
2c010ad
add interactive mode
Browse files
app.py
CHANGED
|
@@ -36,6 +36,8 @@ def chat(user_input, max_new_tokens=512):
|
|
| 36 |
thread.start()
|
| 37 |
|
| 38 |
prompt_len = len(prompt)
|
|
|
|
|
|
|
| 39 |
generated_text = ""
|
| 40 |
for new_text in streamer:
|
| 41 |
generated_text += new_text
|
|
|
|
| 36 |
thread.start()
|
| 37 |
|
| 38 |
prompt_len = len(prompt)
|
| 39 |
+
print(prompt)
|
| 40 |
+
print(prompt_len)
|
| 41 |
generated_text = ""
|
| 42 |
for new_text in streamer:
|
| 43 |
generated_text += new_text
|