sonoisa commited on
Commit
5283d83
1 Parent(s): 390f508

Fix app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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")