Update main.py

main.py

@@ -152,11 +152,11 @@ class Test(BaseModel):
 
 @app.post("/test")
 def test(request: Test):
-            if neg == True:
-                chat = "hello world"
-            elif neg == "True":
-                chat = "Hi there"
-            else:
-                chat = "Hi world"
-        return {'data': f'Prompt is {request.prompt} Model is {request.model} Neg is {request.neg} Chat is {chat}'}
+        if neg == True:
+            chat = "hello world"
+        elif neg == "True":
+            chat = "Hi there"
+        else:
+            chat = "Hi world"
+        return chat