style: run linter

image2text.py

@@ -1,10 +1,10 @@
-import streamlit as st
-import requests
-import numpy as np
 import jax
 import jax.numpy as jnp
-from PIL import Image
+import numpy as np
 import pandas as pd
+import requests
+import streamlit as st
+from PIL import Image
 
 from utils import load_model
 
@@ -25,19 +25,23 @@ def app(model_name):
     query2 = st.text_input("or a URL to an image...")
 
     captions = st.text_input(
-        "Enter candidate captions in comma-separated form.", 
-        value="귀여운 고양이,멋있는 강아지,트랜스포머"
+        "Enter candidate captions in comma-separated form.",
+        value="귀여운 고양이,멋있는 강아지,트랜스포머",
     )
 
     if st.button("질문 (Query)"):
         if not any([query1, query2]):
             st.error("Please upload an image or paste an image URL.")
         else:
-            image_data = query1 if query1 is not None else requests.get(query2, stream=True).raw
+            image_data = (
+                query1 if query1 is not None else requests.get(query2, stream=True).raw
+            )
             image = Image.open(image_data)
             st.image(image)
             captions = captions.split(",")
-            inputs = processor(text=captions, images=image, return_tensors="jax", padding=True)
+            inputs = processor(
+                text=captions, images=image, return_tensors="jax", padding=True
+            )
             inputs["pixel_values"] = jnp.transpose(
                 inputs["pixel_values"], axes=[0, 2, 3, 1]
             )
@@ -48,5 +52,3 @@ def app(model_name):
             st.bar_chart(df)
             # for idx, prob in sorted(enumerate(*probs), key=lambda x: x[1], reverse=True):
             #     st.text(f"Score: `{prob}`, {captions[idx]}")
-
-