Spaces:

mithril-security
/

Santacoder-demo

Runtime error

lauro1 commited on Jun 12, 2023

Commit

7bf2715

1 Parent(s): d105155

demo: add new lines

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ def run_query( prompt):
     try:
         with SecureSession(f"http://{DEMO_SERVER}", POLICY) as secure_session:
             res = secure_session.post(endpoint="/generate", json={"input_text": prompt})
-            return("✅ Query successful\n" + res.text)
     except Exception as err:
         return(f"⛔ Query failed!\n{err}")

     try:
         with SecureSession(f"http://{DEMO_SERVER}", POLICY) as secure_session:
             res = secure_session.post(endpoint="/generate", json={"input_text": prompt})
+            return("✅ Query successful\n" + res.text.replace('\\n', '\n'))
     except Exception as err:
         return(f"⛔ Query failed!\n{err}")