Spaces:

KeerthiVM
/

SkinGPT

Sleeping

KeerthiVM commited on Jun 4

Commit

1914e12

1 Parent(s): 50be4f0

Prompt change

Files changed (1) hide show

SkinGPT.py CHANGED Viewed

@@ -243,6 +243,8 @@ class SkinGPT4(nn.Module):
         if "### Response:" in full_output:
             response = full_output.split("### Response:")[-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()