Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,21 @@
|
|
1 |
import os
|
|
|
2 |
import requests
|
3 |
import gradio as gr
|
4 |
from datetime import datetime
|
5 |
from llama_cpp import Llama # GPT4ALLμ©
|
6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
# βββββββββββββββββββββββββ λͺ¨λΈ λ‘λ© βββββββββββββββββββββββββ
|
8 |
-
llm = Llama(model_path=
|
9 |
|
10 |
# βββββββββββββββββββββββββ NEIS API νΈμΆ ν¨μ βββββββββββββββββββββββββ
|
11 |
def get_school_info(region_code, school_name, api_key):
|
@@ -29,11 +39,8 @@ def generate_gpt_answer(user_question, events):
|
|
29 |
prompt = f"""
|
30 |
### Instruction:
|
31 |
λ€μμ μ리μ΄λ±νκ΅μ 2024λ
7μ νμ¬μΌμ μ
λλ€:
|
32 |
-
|
33 |
{context}
|
34 |
-
|
35 |
μ§λ¬Έ: "{user_question}"
|
36 |
-
|
37 |
μ νμ¬μΌμ μ λ³΄λ§ μ°Έκ³ ν΄μ μ§λ¬Έμ ν΄λΉνλ λ μ§λ₯Ό μ°Ύμ, μμ°μ€λ½κ³ κ°κ²°ν νκ΅μ΄ λ¬Έμ₯μΌλ‘ λ΅ν΄μ£ΌμΈμ. μ: "2024λ
7μ 24μΌμ
λλ€."
|
38 |
### Response:
|
39 |
"""
|
|
|
1 |
import os
|
2 |
+
import subprocess
|
3 |
import requests
|
4 |
import gradio as gr
|
5 |
from datetime import datetime
|
6 |
from llama_cpp import Llama # GPT4ALLμ©
|
7 |
|
8 |
+
# βββββββββββββββββββββββββ λͺ¨λΈ μλ λ€μ΄λ‘λ βββββββββββββββββββββββββ
|
9 |
+
MODEL_PATH = "ggjt-model.bin"
|
10 |
+
GOOGLE_DRIVE_FILE_ID = "1OTd4JOU2ZYC9fU6BP9HZb_8OhCFMq8Ch"
|
11 |
+
|
12 |
+
if not os.path.exists(MODEL_PATH):
|
13 |
+
print("β¬ ggjt-model.bin λ€μ΄λ‘λ μ€...")
|
14 |
+
subprocess.run(["gdown", "--id", GOOGLE_DRIVE_FILE_ID, "-O", MODEL_PATH])
|
15 |
+
print("β
λ€μ΄λ‘λ μλ£.")
|
16 |
+
|
17 |
# βββββββββββββββββββββββββ λͺ¨λΈ λ‘λ© βββββββββββββββββββββββββ
|
18 |
+
llm = Llama(model_path=MODEL_PATH)
|
19 |
|
20 |
# βββββββββββββββββββββββββ NEIS API νΈμΆ ν¨μ βββββββββββββββββββββββββ
|
21 |
def get_school_info(region_code, school_name, api_key):
|
|
|
39 |
prompt = f"""
|
40 |
### Instruction:
|
41 |
λ€μμ μ리μ΄λ±νκ΅μ 2024λ
7μ νμ¬μΌμ μ
λλ€:
|
|
|
42 |
{context}
|
|
|
43 |
μ§λ¬Έ: "{user_question}"
|
|
|
44 |
μ νμ¬μΌμ μ λ³΄λ§ μ°Έκ³ ν΄μ μ§λ¬Έμ ν΄λΉνλ λ μ§λ₯Ό μ°Ύμ, μμ°μ€λ½κ³ κ°κ²°ν νκ΅μ΄ λ¬Έμ₯μΌλ‘ λ΅ν΄μ£ΌμΈμ. μ: "2024λ
7μ 24μΌμ
λλ€."
|
45 |
### Response:
|
46 |
"""
|