Spaces:
Running
Running
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 |
|