Update handler.py

handler.py

@@ -53,7 +53,11 @@ class EndpointHandler:
                 output_ids = self.model.generate(input_ids, max_length=100, temperature=0.3)
 
             json_output = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
-            return json.loads(json_output)
+            # return json.loads(json_output)
+            try:
+                return json.loads(json_output)
+            except:
+                return json_output
 
         except Exception as e:
             return {"error": f"Unexpected error: {str(e)}"}