Spaces:
Sleeping
Sleeping
frankaging
commited on
Commit
·
ec0229c
1
Parent(s):
5f1256d
v3
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ def generate(
|
|
| 89 |
) -> Iterator[str]:
|
| 90 |
|
| 91 |
# tokenize and prepare the input
|
| 92 |
-
prompt = prompt_no_input_template % message
|
| 93 |
prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 94 |
|
| 95 |
unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
|
|
@@ -144,8 +144,7 @@ chat_interface = gr.ChatInterface(
|
|
| 144 |
stop_btn=None,
|
| 145 |
examples=[
|
| 146 |
["who are you?"],
|
| 147 |
-
["
|
| 148 |
-
["Teach me how to program in python."],
|
| 149 |
],
|
| 150 |
)
|
| 151 |
|
|
|
|
| 89 |
) -> Iterator[str]:
|
| 90 |
|
| 91 |
# tokenize and prepare the input
|
| 92 |
+
prompt = prompt_no_input_template % message.lower()
|
| 93 |
prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 94 |
|
| 95 |
unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
|
|
|
|
| 144 |
stop_btn=None,
|
| 145 |
examples=[
|
| 146 |
["who are you?"],
|
| 147 |
+
["Can you tell me how to improve my health?"],
|
|
|
|
| 148 |
],
|
| 149 |
)
|
| 150 |
|