The sample code above would raise an error if the model returned improperly formatted responses before. Now it will show the actual response so you can see why re failed to parse it and it will return an object that is {'Software/System': 'Unknown', 'Issue/Request': 'Unknown'} instead of an error.

README.md

@@ -64,7 +64,24 @@ class SITCC_T5_Classifier:
     Returns:
     dict: A dictionary containing the software/system and issue/request.
     """
+    # handle null input
+    if not response:
+      print("Error: Response is empty")
+      return {
+        "Software/System": "Unknown",
+        "Issue/Request": "Unknown"
+      }
+
     matches = re.search(r'Software/System: (.*) Issue/Request: (.*)</s>', response, re.DOTALL)
+    
+    # handle improperly formatted responses
+    # TODO: Provide more robust improper format handing
+    if not matches or len(matches.groups()) < 2:
+      print(f"Error: Response format does not match expected format \"Software/System: (.*) Issue/Request: (.*)</s>\"\nResponse: \"{response}\" ")
+      return {
+        "Software/System": "Unknown",
+        "Issue/Request": "Unknown"
+      }
     return {
       "Software/System": matches.group(1),
       "Issue/Request": matches.group(2)