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 |
"""
|