Pedro Cuenca
commited on
Commit
路
d6e9f87
1
Parent(s):
3500e67
Half the number of requests.
Browse filesThis requires a corresponding change in the backend.
- app/streamlit/app.py +4 -4
- 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
|
8 |
|
9 |
st.sidebar.markdown(
|
10 |
"""
|
@@ -67,7 +67,9 @@ if prompt != "":
|
|
67 |
|
68 |
try:
|
69 |
backend_url = st.secrets["BACKEND_SERVER"] + "/generate"
|
70 |
-
|
|
|
|
|
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 |
-
|
|
|
19 |
images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
|
20 |
-
|
|
|
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 |
|