Pedro Cuenca commited on
Commit
13caa60
·
1 Parent(s): f58c732

Display info about model and run, below predictions.

Browse files
Files changed (1) hide show
  1. app/streamlit/app.py +8 -0
app/streamlit/app.py CHANGED
@@ -77,6 +77,14 @@ if prompt != "":
77
  cols[(i % n_columns) * 2].image(img)
78
  container.markdown(f"**{prompt}**")
79
 
 
 
 
 
 
 
 
 
80
  st.button("Again!", key="again_button")
81
 
82
  except ServiceError as error:
 
77
  cols[(i % n_columns) * 2].image(img)
78
  container.markdown(f"**{prompt}**")
79
 
80
+ version_url = st.secrets["BACKEND_SERVER"] + "/version"
81
+ version = get_model_version(version_url)
82
+ st.sidebar.markdown(f"<small><center>{version}</center></small>", unsafe_allow_html=True)
83
+
84
+ st.markdown(f"""
85
+ These results have been obtained using model `{version}` from [an ongoing training run](https://wandb.ai/dalle-mini/dalle-mini/runs/mheh9e55).
86
+ """)
87
+
88
  st.button("Again!", key="again_button")
89
 
90
  except ServiceError as error: