Spaces:
Paused
Paused
harpreetsahota
commited on
Commit
·
0d4a6a6
1
Parent(s):
9a81d74
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ def generate_model_response(message: str) -> str:
|
|
61 |
temperature=0.1,
|
62 |
streamer=streamer
|
63 |
)
|
64 |
-
|
65 |
|
66 |
def extract_response_content(full_response: str) -> str:
|
67 |
response_start_index = full_response.find("### Assistant:")
|
|
|
61 |
temperature=0.1,
|
62 |
streamer=streamer
|
63 |
)
|
64 |
+
return tokenizer.decode(output[0], skip_special_tokens=True)
|
65 |
|
66 |
def extract_response_content(full_response: str) -> str:
|
67 |
response_start_index = full_response.find("### Assistant:")
|