Update app.py

app.py

@@ -10,7 +10,7 @@ import warnings
 warnings.filterwarnings("ignore", category=UserWarning)
 
 # model_name = "AI-Sweden-Models/gpt-sw3-126m-instruct"
-model_name = "AI-Sweden-Models/gpt-sw3-356m-instruct"
+model_name = "AI-Sweden-Models/gpt-sw3-126m-instruct"
 
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -78,7 +78,7 @@ class StopOnTokenCriteria(StoppingCriteria):
 stop_on_token_criteria = StopOnTokenCriteria(stop_token_id=tokenizer.bos_token_id)
 
 st.title("Paralegal Assistant")
-st.caption("RAG: föräldrabalken")
+st.subheader("RAG: föräldrabalken")
 
 # Initialize chat history
 if "messages" not in st.session_state: