tori29umai commited on
Commit
fb0db39
·
verified ·
1 Parent(s): 0ef21fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -21,6 +21,20 @@ MODEL_FILE_EXTENSION = '.gguf'
21
  BASE_PATH = os.path.dirname(os.path.abspath(__file__))
22
  MODEL_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
23
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  class ConfigManager:
25
  @staticmethod
26
  def load_settings(filename):
 
21
  BASE_PATH = os.path.dirname(os.path.abspath(__file__))
22
  MODEL_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
23
 
24
+
25
+ # モデルディレクトリが存在しない場合は作成
26
+ if not os.path.exists("models"):
27
+ os.makedirs("models")
28
+
29
+ # 使用するモデルのファイル名を指定
30
+ model_filename = "EZO-Common-9B-gemma-2-it.f16.gguf.gguf"
31
+ model_path = os.path.join("models", model_filename)
32
+
33
+ # モデルファイルが存在しない場合はダウンロード
34
+ if not os.path.exists(model_path):
35
+ dl_guff_model("models", f"https://huggingface.co/Aratako/Ninja-v1-RP-expressive-v2-GGUF/resolve/main/{model_filename}")
36
+
37
+
38
  class ConfigManager:
39
  @staticmethod
40
  def load_settings(filename):