Pedro Cuenca commited on
Commit
d6e9f87
1 Parent(s): 3500e67

Half the number of requests.

Browse files

This requires a corresponding change in the backend.

Files changed (2) hide show
  1. app/streamlit/app.py +4 -4
  2. app/streamlit/backend.py +4 -2
app/streamlit/app.py CHANGED
@@ -4,7 +4,7 @@
4
  from datetime import datetime
5
 
6
  import streamlit as st
7
- from backend import ServiceError, get_images_from_backend, get_model_version
8
 
9
  st.sidebar.markdown(
10
  """
@@ -67,7 +67,9 @@ if prompt != "":
67
 
68
  try:
69
  backend_url = st.secrets["BACKEND_SERVER"] + "/generate"
70
- selected = get_images_from_backend(prompt, backend_url)
 
 
71
 
72
  margin = 0.1 # for better position of zoom in arrow
73
  n_columns = 3
@@ -76,8 +78,6 @@ if prompt != "":
76
  cols[(i % n_columns) * 2].image(img)
77
  container.markdown(f"**{prompt}**")
78
 
79
- version_url = st.secrets["BACKEND_SERVER"] + "/version"
80
- version = get_model_version(version_url)
81
  st.sidebar.markdown(
82
  f"<small><center>{version}</center></small>", unsafe_allow_html=True
83
  )
 
4
  from datetime import datetime
5
 
6
  import streamlit as st
7
+ from backend import ServiceError, get_images_from_backend
8
 
9
  st.sidebar.markdown(
10
  """
 
67
 
68
  try:
69
  backend_url = st.secrets["BACKEND_SERVER"] + "/generate"
70
+ response = get_images_from_backend(prompt, backend_url)
71
+ selected = response["images"]
72
+ version = response["version"]
73
 
74
  margin = 0.1 # for better position of zoom in arrow
75
  n_columns = 3
 
78
  cols[(i % n_columns) * 2].image(img)
79
  container.markdown(f"**{prompt}**")
80
 
 
 
81
  st.sidebar.markdown(
82
  f"<small><center>{version}</center></small>", unsafe_allow_html=True
83
  )
app/streamlit/backend.py CHANGED
@@ -15,9 +15,11 @@ class ServiceError(Exception):
15
  def get_images_from_backend(prompt, backend_url):
16
  r = requests.post(backend_url, json={"prompt": prompt})
17
  if r.status_code == 200:
18
- images = r.json()["images"]
 
19
  images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
20
- return images
 
21
  else:
22
  raise ServiceError(r.status_code)
23
 
 
15
  def get_images_from_backend(prompt, backend_url):
16
  r = requests.post(backend_url, json={"prompt": prompt})
17
  if r.status_code == 200:
18
+ json = r.json()
19
+ images = json["images"]
20
  images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
21
+ version = json.get("version", "unknown")
22
+ return {"images": images, "version": version}
23
  else:
24
  raise ServiceError(r.status_code)
25