fix loading

app.py

@@ -820,24 +820,22 @@ def main():
                 LOGGER.info(f"FORM {generation_config=}")
 
         with st.expander("Info"):
-            try:
-                data_card = dataset_info(st.session_state.dataset_name).cardData
-            except (HFValidationError, RepositoryNotFoundError):
-                pass
-            else:
-                st.caption("Dataset")
-                st.write(data_card)
-            try:
-                model_info_respose = model_info(model)
-                model_card = model_info_respose.cardData
-                st.session_state["generation_config"]["is_chat"] = (
-                    "conversational" in model_info_respose.tags
-                )
-            except (HFValidationError, RepositoryNotFoundError):
-                pass
-            else:
-                st.caption("Model")
-                st.write(model_card)
+            if "dataset_name" in st.session_state:
+                try:
+                    data_card = dataset_info(st.session_state.dataset_name).cardData
+                    st.caption("Dataset")
+                    st.write(data_card)
+                except (HFValidationError, RepositoryNotFoundError):
+                    pass
+
+            if "api_call_function" in st.session_state:
+                try:
+                    model_info_response = model_info(model)
+                    model_card = model_info_response.cardData
+                    st.caption("Model")
+                    st.write(model_card)
+                except (HFValidationError, RepositoryNotFoundError):
+                    pass
 
             # st.write(f"Model max length: {AutoTokenizer.from_pretrained(model).model_max_length}")