Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,12 @@
|
|
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 |
# 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"
|