Spaces:

harir
/

Review-Toxicity-Checker

Sleeping

harir commited on Apr 24, 2024

Commit

150178c

verified ·

1 Parent(s): f5a4353

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -133,7 +133,7 @@ def revise_review(review, api_key, model_id, highlight_color):
                     review_score = 1
                     revision_count +=1
                     revision = query_model_revise(sentence, api_key, model_id, revision_prompt)
-                    revision = extract_text_in_quotes(revision)
                     revision = revision.strip().strip('"')
                     review_revision += f" <div style='background-color: {highlight_color}; display: inline;'>{revision}</div>"
             else:

                     review_score = 1
                     revision_count +=1
                     revision = query_model_revise(sentence, api_key, model_id, revision_prompt)
+                    revision = extract_quotes(revision)
                     revision = revision.strip().strip('"')
                     review_revision += f" <div style='background-color: {highlight_color}; display: inline;'>{revision}</div>"
             else: