Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -132,8 +132,10 @@ async def echo(websocket):
|
|
132 |
)["answer"]
|
133 |
if response.startswith("? "):
|
134 |
response.replace("? ", "")
|
135 |
-
if
|
136 |
-
response.replace("Assistant:
|
|
|
|
|
137 |
await websocket.send(json.dumps({"response": response}))
|
138 |
|
139 |
async def main():
|
|
|
132 |
)["answer"]
|
133 |
if response.startswith("? "):
|
134 |
response.replace("? ", "")
|
135 |
+
if "assistant: " in response.lower() or "ai: " in response.lower():
|
136 |
+
response.replace("Assistant:", "").replace("AI:", "")
|
137 |
+
response.replace("\n", "")
|
138 |
+
response.strip()
|
139 |
await websocket.send(json.dumps({"response": response}))
|
140 |
|
141 |
async def main():
|