diff --git a/README.md b/README.md
index ddce082b70..c398140d1a 100644
--- a/README.md
+++ b/README.md
@@ -123,22 +123,22 @@ For more information run `python3 torchchat.py generate --help`
 
 ### Browser
 This mode provides access to the model via the browser's localhost.
+
+Launch an interactive chat with your model. Running the command will automatically open a tab in your browser. [Streamlit](https://streamlit.io/) should already be installed by the `install_requirements.sh` script.
+```
+streamlit run torchchat.py -- browser <model_name> <model_args>
+```
+
+For example, to quantize and chat with LLaMA3:
 [skip default]: begin
 ```
-python3 torchchat.py browser llama3
+streamlit run torchchat.py -- browser llama3 --quantize '{"precision": {"dtype":"float16"}, "executor":{"accelerator":"cpu"}}' --max-new-tokens 256 --compile
 ```
 [skip default]: end
 
 
-*Running on http://127.0.0.1:5000* should be printed out on the
- terminal. Click the link or go to
- [http://127.0.0.1:5000](http://127.0.0.1:5000) on your browser to
- start interacting with it.
 
-Enter some text in the input box, then hit the enter key or click the
-“SEND” button. After a second or two, the text you entered together
-with the generated text will be displayed. Repeat to have a
-conversation.
+
 
 
 
diff --git a/browser/browser.py b/browser/browser.py
index 5c3fca797e..0074cf3924 100644
--- a/browser/browser.py
+++ b/browser/browser.py
@@ -4,40 +4,88 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
-import subprocess
-import sys
+import time
+
+import streamlit as st
+from api.api import CompletionRequest, OpenAiApiGenerator
+
+from build.builder import BuilderArgs, TokenizerArgs
+
+from generate import GeneratorArgs
 
 
 def main(args):
+    builder_args = BuilderArgs.from_args(args)
+    speculative_builder_args = BuilderArgs.from_speculative_args(args)
+    tokenizer_args = TokenizerArgs.from_args(args)
+    generator_args = GeneratorArgs.from_args(args)
+    generator_args.chat_mode = False
+
+    @st.cache_resource
+    def initialize_generator() -> OpenAiApiGenerator:
+        return OpenAiApiGenerator(
+            builder_args,
+            speculative_builder_args,
+            tokenizer_args,
+            generator_args,
+            args.profile,
+            args.quantize,
+            args.draft_quantize,
+        )
+
+    gen = initialize_generator()
+
+    st.title("torchchat")
+
+    # Initialize chat history
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Display chat messages from history on app rerun
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # Accept user input
+    if prompt := st.chat_input("What is up?"):
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        # Display user message in chat message container
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        # Display assistant response in chat message container
+        with st.chat_message("assistant"), st.status(
+            "Generating... ", expanded=True
+        ) as status:
+
+            req = CompletionRequest(
+                model=gen.builder_args.checkpoint_path,
+                prompt=prompt,
+                temperature=generator_args.temperature,
+                messages=[],
+            )
+
+            def unwrap(completion_generator):
+                start = time.time()
+                tokcount = 0
+                for chunk_response in completion_generator:
+                    content = chunk_response.choices[0].delta.content
+                    if not gen.is_llama3_model or content not in set(
+                        gen.tokenizer.special_tokens.keys()
+                    ):
+                        yield content
+                    if content == gen.tokenizer.eos_id():
+                        yield "."
+                    tokcount += 1
+                status.update(
+                    label="Done, averaged {:.2f} tokens/second".format(
+                        tokcount / (time.time() - start)
+                    ),
+                    state="complete",
+                )
+
+            response = st.write_stream(unwrap(gen.completion(req)))
 
-    # Directory Containing the server file "chat_in_browser.py"
-    server_dir = "browser"
-
-    # Look for port from cmd args. Default to 5000 if not found.
-    port = 5000
-    i = 2
-    while i < len(sys.argv):
-        if sys.argv[i] == "--port":
-            if i + 1 < len(sys.argv):
-                # Extract the value and remove '--port' and the value from sys.argv
-                port = sys.argv[i + 1]
-                del sys.argv[i : i + 2]
-                break
-        else:
-            i += 1
-
-    # Construct arguments for the flask app minus 'browser' command
-    # plus '--chat'
-    args_plus_chat = ["'{}'".format(s) for s in sys.argv[1:] if s != "browser"] + [
-        '"--chat"'
-    ]
-    formatted_args = ", ".join(args_plus_chat)
-    command = [
-        "flask",
-        "--app",
-        f"{server_dir}/chat_in_browser:create_app(" + formatted_args + ")",
-        "run",
-        "--port",
-        f"{port}",
-    ]
-    subprocess.run(command)
+        # Add assistant response to chat history
+        st.session_state.messages.append({"role": "assistant", "content": response})
diff --git a/browser/chat_in_browser.py b/browser/chat_in_browser.py
deleted file mode 100644
index e835fa0090..0000000000
--- a/browser/chat_in_browser.py
+++ /dev/null
@@ -1,107 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
-import subprocess
-
-from flask import Flask, render_template, request
-
-convo = ""
-disable_input = False
-
-
-def create_app(*args):
-    app = Flask(__name__)
-
-    # create a new process and set up pipes for communication
-    proc = subprocess.Popen(
-        ["python3", "generate.py", *args], stdin=subprocess.PIPE, stdout=subprocess.PIPE
-    )
-
-    @app.route("/")
-    def main():
-        print("Starting chat session.")
-        line = b""
-        output = ""
-        global disable_input
-
-        while True:
-            buffer = proc.stdout.read(1)
-            line += buffer
-            try:
-                decoded = line.decode("utf-8")
-            except:
-                continue
-
-            if decoded.endswith("Do you want to enter a system prompt? Enter y for yes and anything else for no. \n"):
-                print(f"| {decoded}")
-                proc.stdin.write("\n".encode("utf-8"))
-                proc.stdin.flush()
-                line = b""
-            elif line.decode("utf-8").startswith("User: "):
-                print(f"| {decoded}")
-                break
-
-            if decoded.endswith("\r") or decoded.endswith("\n"):
-                decoded = decoded.strip()
-                print(f"| {decoded}")
-                output += decoded + "\n"
-                line = b""
-
-        return render_template(
-            "chat.html",
-            convo="Hello! What is your prompt?",
-            disable_input=disable_input,
-        )
-
-    @app.route("/chat", methods=["GET", "POST"])
-    def chat():
-        # Retrieve the HTTP POST request parameter value from
-        # 'request.form' dictionary
-        _prompt = request.form.get("prompt", "")
-        proc.stdin.write((_prompt + "\n").encode("utf-8"))
-        proc.stdin.flush()
-
-        print(f"User: {_prompt}")
-
-        line = b""
-        output = ""
-        global disable_input
-
-        while True:
-            buffer = proc.stdout.read(1)
-            line += buffer
-            try:
-                decoded = line.decode("utf-8")
-            except:
-                continue
-
-            if decoded.startswith("User: "):
-                break
-            if decoded.startswith("=========="):
-                disable_input = True
-                break
-            if decoded.endswith("\r") or decoded.endswith("\n"):
-                decoded = decoded.strip()
-                print(f"| {decoded}")
-                output += decoded + "\n"
-                line = b""
-
-        # Strip "Model: " from output
-        model_prefix = "Model: "
-        if output.startswith(model_prefix):
-            output = output[len(model_prefix) :]
-        else:
-            print("But output is", output)
-
-        global convo
-
-        if _prompt:
-            convo += "<H1>User</H1>\n<p> " + _prompt + " </p>\n\n"
-            convo += "<H1>Model</H1>\n<p> " + output + " </p>\n\n"
-
-        return render_template("chat.html", convo=convo, disable_input=disable_input)
-
-    return app
diff --git a/browser/static/css/style.css b/browser/static/css/style.css
deleted file mode 100644
index 61c26cd4c8..0000000000
--- a/browser/static/css/style.css
+++ /dev/null
@@ -1,96 +0,0 @@
-body {
-  font-family: "Helvetica Neue", Arial, sans-serif;
-  background-color: #f5f5f5;
-  color: #333;
-  margin: 0;
-  padding: 0;
-}
-
-.main-content {
-  width: 66%;
-  padding: 20px;
-  float: left; /* Add this line */
-}
-
-.main-content h1 {
-  font-family: "Helvetica Neue", Arial, sans-serif;
-  font-size: 0.8em; /* Adjust as needed */
-  color: rgba(208, 0, 0, 0.5); /* Semi-transparent PyTorch red */
-  margin-left: 20px; /* Align with the left side of the text box */
-  margin-bottom: 2pt; /* Adjust as needed */
-}
-
-pre {
-  background-color: #fff;
-  border: 1px solid #ddd;
-  border-radius: 5px;
-  padding: 10px;
-  margin: 20px;
-  font-size: 16px;
-  line-height: 1.6;
-  overflow: auto;
-}
-
-form {
-  background-color: #fff;
-  border: 1px solid #ddd;
-  border-radius: 5px;
-  padding: 20px;
-  margin: 20px;
-}
-
-label {
-  font-weight: bold;
-  color: #555;
-}
-
-input[type="text"] {
-  border: 1px solid #ddd;
-  border-radius: 5px;
-  padding: 10px;
-  width: 100%;
-  box-sizing: border-box;
-  margin-bottom: 20px;
-}
-
-input[type="submit"] {
-  background-color: #D00000; /* PyTorch red */
-  color: #fff;
-  border: none;
-  border-radius: 5px;
-  padding: 10px 20px;
-  cursor: pointer;
-  text-transform: uppercase;
-}
-
-input[type="submit"]:hover {
-  background-color: #b30000;
-}
-
-.logo-container {
-  width: 33%;
-  position: fixed;
-  right: 0;
-  text-align: center;
-  padding: 20px;
-}
-
-.logo {
-  width: 100%;
-  max-width: 300px; /* Adjust as needed */
-}
-
-.chat-title {
-  color: #D00000; /* PyTorch red */
-  font-size: 3em; /* Adjust as needed */
-}
-
-p {
-  font-family: "Helvetica Neue", Arial, sans-serif;
-  background-color: #fff;
-  border: 1px solid #ddd;
-  border-radius: 15px; /* Rounded edges */
-  padding: 10px;
-  margin: 20px;
-  color: #D00000; /* PyTorch red */
-}
diff --git a/browser/templates/chat.html b/browser/templates/chat.html
deleted file mode 100644
index 36ba086c82..0000000000
--- a/browser/templates/chat.html
+++ /dev/null
@@ -1,27 +0,0 @@
-<!DOCTYPE html>
-<html lang="en">
-<head>
-  <meta charset="utf-8">
-  <title>torchchat</title>
-  <link rel="stylesheet" type="text/css" href="{{ url_for('static', filename='css/style.css') }}">
-</head>
-
-<body>
-  <div class="main-content">
-    {{ convo | safe }}
-    <form action="chat" method="post">
-      <label for="username">Prompt: </label>
-      <input type="text" id="prompt" name="prompt"><br>
-      <input type="submit" value="SEND">
-    </form>
-  </div>
-<div class="logo-container">
-  <!--
-  <img src="{{ url_for('static', filename='https://upload.wikimedia.org/wikipedia/commons/1/10/PyTorch_logo_icon.svg') }}" alt="PyTorch Logo" class="logo">
-  -->
-  <img src="https://upload.wikimedia.org/wikipedia/commons/1/10/PyTorch_logo_icon.svg" alt="PyTorch Logo" class="logo">
-  <h1 class="chat-title">torchchat</h1>
-</div>
-
-</body>
-
diff --git a/cli.py b/cli.py
index 603d02dbea..ede80942b8 100644
--- a/cli.py
+++ b/cli.py
@@ -134,12 +134,6 @@ def add_arguments_for_verb(parser, verb: str) -> None:
         default=None,
         help="Profile path.",
     )
-    parser.add_argument(
-        "--port",
-        type=int,
-        default=5000,
-        help="Port for the web server in browser mode",
-    )
     _add_cli_metadata_args(parser)
 
     # WIP Features (suppressed from --help)
@@ -150,7 +144,10 @@ def add_arguments_for_verb(parser, verb: str) -> None:
 
 # Add CLI Args representing user provided exported model files
 def _add_export_output_path_args(parser) -> None:
-    output_path_parser = parser.add_argument_group("Export Output Path Args", "Specify the output path for the exported model files")
+    output_path_parser = parser.add_argument_group(
+        "Export Output Path Args",
+        "Specify the output path for the exported model files",
+    )
     output_path_parser.add_argument(
         "--output-pte-path",
         type=str,
@@ -167,7 +164,10 @@ def _add_export_output_path_args(parser) -> None:
 
 # Add CLI Args representing user provided exported model files
 def _add_exported_input_path_args(parser) -> None:
-    exported_model_path_parser = parser.add_argument_group("Exported Model Path Args", "Specify the path of the exported model files to ingest")
+    exported_model_path_parser = parser.add_argument_group(
+        "Exported Model Path Args",
+        "Specify the path of the exported model files to ingest",
+    )
     exported_model_path_parser.add_argument(
         "--dso-path",
         type=Path,
@@ -228,7 +228,9 @@ def _configure_artifact_inventory_args(parser, verb: str) -> None:
 
 # Add CLI Args specific to user prompted generation
 def _add_generation_args(parser) -> None:
-    generator_parser = parser.add_argument_group("Generation Args", "Configs for generating output based on provided prompt")
+    generator_parser = parser.add_argument_group(
+        "Generation Args", "Configs for generating output based on provided prompt"
+    )
     generator_parser.add_argument(
         "--prompt",
         type=str,
@@ -275,7 +277,9 @@ def _add_generation_args(parser) -> None:
 
 # Add CLI Args specific to Model Evaluation
 def _add_evaluation_args(parser) -> None:
-    eval_parser = parser.add_argument_group("Evaluation Args", "Configs for evaluating model performance")
+    eval_parser = parser.add_argument_group(
+        "Evaluation Args", "Configs for evaluating model performance"
+    )
     eval_parser.add_argument(
         "--tasks",
         nargs="+",
diff --git a/requirements.txt b/requirements.txt
index 25e3d02149..512f0bc7ab 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,4 +25,4 @@ ninja
 zstd
 
 # Browser mode
-flask
+streamlit