Spaces:

KeerthiVM
/

SkinGPT

Sleeping

KeerthiVM commited on Jun 5

Commit

5b1c0ec

1 Parent(s): 2096474

Prompt change

Files changed (1) hide show

SkinGPT.py CHANGED Viewed

@@ -249,8 +249,8 @@ class SkinGPT4(nn.Module):
         if "### Response:" in full_output:
             response = full_output.split("### Response:")[-1].strip()
-        # elif "assistant:" in full_output:
-        #     response = full_output.split("assistant:")[-1].strip()
         else:
             response = full_output.strip()

         if "### Response:" in full_output:
             response = full_output.split("### Response:")[-1].strip()
+        elif "assistant:" in full_output:
+            response = full_output.split("assistant:")[-1].strip()
         else:
             response = full_output.strip()