Ari1020 commited on
Commit
5ed9825
·
verified ·
1 Parent(s): 340217a

Correct flag mistakes

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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