Increased token limit

SkinGPT.py

@@ -239,14 +239,24 @@ class SkinGPT4(nn.Module):
         # response = full_output.split("### Response:")[-1].strip()
 
         print("Full output:", full_output)
-        print("Split parts:", full_output.split("### Response:"))
-        # response = full_output.split("### Response:")[-1].strip()
 
-        response_parts = full_output.split("### Response:")
-        response = response_parts[-1].strip() if len(response_parts) > 1 else full_output
+        if "### Response:" in full_output:
+            response = full_output.split("### Response:")[-1].strip()
+        else:
+            response = full_output.strip()
+
+        response = response.strip('"').strip()
 
-        print("Final response:", response_parts)
+        print("Processed response:", response)  # Debug print
         return response
+        # print("Split parts:", full_output.split("### Response:"))
+        # # response = full_output.split("### Response:")[-1].strip()
+        #
+        # response_parts = full_output.split("### Response:")
+        # response = response_parts[-1].strip() if len(response_parts) > 1 else full_output
+        #
+        # print("Final response:", response_parts)
+        # return response
 
         # return response