Fix text generation

lib/api.py

@@ -14,10 +14,11 @@ from .config import config
 
 
 def txt2txt_generate(api_key, service, parameters, **kwargs):
+    model = parameters.get("model", "")
     base_url = config.services[service].url
 
     if service == "hf":
-        base_url = f"{base_url}/{parameters['model']}/v1"
+        base_url = f"{base_url}/{model}/v1"
 
     try:
         if service == "anthropic":

pages/1_💬_Text_Generation.py

@@ -203,7 +203,7 @@ if prompt := st.chat_input(
     with st.chat_message("assistant"):
         session_key = f"api_key_{service}"
         api_key = st.session_state[session_key] or text_services[service].api_key
-        response = txt2txt_generate(api_key, service, model, parameters)
+        response = txt2txt_generate(api_key, service, parameters)
         st.session_state.running = False
 
     st.session_state.txt2txt_messages.append({"role": "user", "content": prompt})