Style: run black.

app/streamlit/app.py

@@ -79,11 +79,15 @@ if prompt != "":
 
         version_url = st.secrets["BACKEND_SERVER"] + "/version"
         version = get_model_version(version_url)
-        st.sidebar.markdown(f"<small><center>{version}</center></small>", unsafe_allow_html=True)
+        st.sidebar.markdown(
+            f"<small><center>{version}</center></small>", unsafe_allow_html=True
+        )
 
-        st.markdown(f"""
+        st.markdown(
+            f"""
         These results have been obtained using model `{version}` from [an ongoing training run](https://wandb.ai/dalle-mini/dalle-mini/runs/mheh9e55).
-        """)
+        """
+        )
 
         st.button("Again!", key="again_button")
 

app/streamlit/backend.py

@@ -5,15 +5,14 @@ from io import BytesIO
 import base64
 from PIL import Image
 
+
 class ServiceError(Exception):
     def __init__(self, status_code):
         self.status_code = status_code
 
+
 def get_images_from_backend(prompt, backend_url):
-    r = requests.post(
-        backend_url, 
-        json={"prompt": prompt}
-    )
+    r = requests.post(backend_url, json={"prompt": prompt})
     if r.status_code == 200:
         images = r.json()["images"]
         images = [Image.open(BytesIO(base64.b64decode(img))) for img in images]
@@ -21,6 +20,7 @@ def get_images_from_backend(prompt, backend_url):
     else:
         raise ServiceError(r.status_code)
 
+
 def get_model_version(url):
     r = requests.get(url)
     if r.status_code == 200: