Spaces:
Sleeping
Sleeping
Correct flag mistakes
Browse files
app.py
CHANGED
|
@@ -49,6 +49,12 @@ def respond(
|
|
| 49 |
token = message.choices[0].delta.content
|
| 50 |
|
| 51 |
response += token
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
yield response
|
| 53 |
|
| 54 |
|
|
|
|
| 49 |
token = message.choices[0].delta.content
|
| 50 |
|
| 51 |
response += token
|
| 52 |
+
|
| 53 |
+
import difflib
|
| 54 |
+
close_matches = difflib.get_close_matches(flag, response.split(), n=1, cutoff=0.6)
|
| 55 |
+
if close_matches:
|
| 56 |
+
response = response.replace(close_matches[0], flag)
|
| 57 |
+
|
| 58 |
yield response
|
| 59 |
|
| 60 |
|