Spaces:

Ping9gu
/

my-diary

Sleeping

Ping9gu commited on 20 days ago

Commit

6195808

1 Parent(s): 4d20401

Change to smaller Korean language model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ def generate_diary(keywords):
             return "키워드를 입력해주세요."
         client = InferenceClient(
-            model="nlpai-lab/kullm-polyglot-5.8b-v2",
             token=os.environ.get("HUGGINGFACE_API_KEY")
         )

             return "키워드를 입력해주세요."
         client = InferenceClient(
+            model="EleutherAI/polyglot-ko-1.3b",
             token=os.environ.get("HUGGINGFACE_API_KEY")
         )