diff --git a/prescribing-info-app/main.py b/prescribing-info-app/main.py
index c31205c..713f97f 100644
--- a/prescribing-info-app/main.py
+++ b/prescribing-info-app/main.py
@@ -37,9 +37,10 @@ def update_drug_data(state: WriterState, drug_name: str) -> None:
     state["prescribing_info_summary"] = ""
 
     for chunk in stream_complete(
-        prescribing_summary_prompt.format(
+        initial_text=prescribing_summary_prompt.format(
             prescribing_details=state["raw_prescribing_info"]
-        )
+        ),
+        config={"model": "palmyra-med"}
     ):
         state["prescribing_info_summary"] += chunk
 
@@ -174,4 +175,4 @@ def contributing_sources_change_vis(state: WriterState) -> None:
     }
 )
 
-initial_state.import_stylesheet("style", "/static/custom.css?1")
+initial_state.import_stylesheet("style", "/static/custom.css")