Update app.py

app.py

@@ -138,21 +138,15 @@ def generate_arctic_response():
         abort_chat(f"Conversation length too long. Please keep it under {max_tokens} tokens.")
     
     st.session_state.messages.append({"role": "assistant", "content": ""})
-    for event_index, event in enumerate(replicate.stream("snowflake/snowflake-arctic-instruct",
+    for event in replicate.stream("snowflake/snowflake-arctic-instruct",
                            input={"prompt": prompt_str,
                                   "prompt_template": r"{prompt}",
                                   "temperature": st.session_state.temperature,
                                   "top_p": st.session_state.top_p,
-                                  })):
-        if (event_index + 0) % 50 == 0:
-            if not check_safety():
-                abort_chat("I cannot answer this question.")
-        st.session_state.messages[-1]["content"] += str(event)
-        yield str(event)
-
-    # Final safety check...
-    if not check_safety():
-        abort_chat("I cannot answer this question.")
+                                  }):
+    print(str(event), end="")
+
+
 
 if __name__ == "__main__":
     main()