isonuma commited on
Commit
4e5672e
1 Parent(s): 57e364c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import streamlit as st
2
- from utils import is_hiragana_or_katakana
3
  import torch
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, LlamaTokenizer
5
  from st_keyup import st_keyup
6
 
 
 
 
7
  # model_name_or_path = "tokyotech-llm/Llama-3-Swallow-8B-v0.1"
8
  model_name_or_path = "tokyotech-llm/Swallow-7b-hf"
9
  # model_name_or_path = "llm-jp/llm-jp-1.3b-v1.0"
 
1
  import streamlit as st
2
+
3
  import torch
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, LlamaTokenizer
5
  from st_keyup import st_keyup
6
 
7
+ from utils import is_hiragana_or_katakana, search_candidates
8
+
9
+
10
  # model_name_or_path = "tokyotech-llm/Llama-3-Swallow-8B-v0.1"
11
  model_name_or_path = "tokyotech-llm/Swallow-7b-hf"
12
  # model_name_or_path = "llm-jp/llm-jp-1.3b-v1.0"