Fix app.py
Browse files
app.py
CHANGED
@@ -320,9 +320,9 @@ description_text = st.empty()
|
|
320 |
if "model" not in st.session_state:
|
321 |
description_text.text("日本語CLIPモデル読み込み中... ")
|
322 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
323 |
-
text_model = ClipTextModel("sonoisa/clip-vit-b-32-japanese-v1", device=device)
|
324 |
-
model = ClipModel("sonoisa/clip-vit-b-32-japanese-v1", device=device)
|
325 |
# model = DummyClipModel(text_model)
|
|
|
326 |
st.session_state.model = model
|
327 |
|
328 |
print("extract dataset")
|
|
|
320 |
if "model" not in st.session_state:
|
321 |
description_text.text("日本語CLIPモデル読み込み中... ")
|
322 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
323 |
+
# text_model = ClipTextModel("sonoisa/clip-vit-b-32-japanese-v1", device=device)
|
|
|
324 |
# model = DummyClipModel(text_model)
|
325 |
+
model = ClipModel("sonoisa/clip-vit-b-32-japanese-v1", device=device)
|
326 |
st.session_state.model = model
|
327 |
|
328 |
print("extract dataset")
|