update: responses and safety in proper json format

app.py

@@ -68,7 +68,9 @@ def inference(prompt: str, max_tokens: int = 256) -> str:
         skip_special_tokens=True
     )
     safety = check_safety(response)
-    return response, safety
+
+    json = {"response": response, "safety": safety}
+    return json
     
     # # Use TextIteratorStreamer instead of TextStreamer
     # streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)