arxivgpt kim commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,12 +17,9 @@ sys.excepthook = exception_handler
|
|
| 17 |
sys.tracebacklimit = 0
|
| 18 |
|
| 19 |
def predict(inputs):
|
| 20 |
-
# OpenAI APIμ μμ²μ λ³΄λΌ νμ΄λ‘λλ₯Ό ꡬμ±ν©λλ€.
|
| 21 |
payload = {
|
| 22 |
-
"model": "gpt-
|
| 23 |
-
"
|
| 24 |
-
"temperature": 0.7,
|
| 25 |
-
"max_tokens": 150
|
| 26 |
}
|
| 27 |
|
| 28 |
headers = {
|
|
@@ -30,22 +27,13 @@ def predict(inputs):
|
|
| 30 |
"Authorization": f"Bearer {OPENAI_API_KEY}"
|
| 31 |
}
|
| 32 |
|
| 33 |
-
|
| 34 |
-
# νμΈν μ¬ν: μμ²μ 보λ΄λ μ½λ λΆλΆμ΄ μ¬λ°λ₯Έκ°?
|
| 35 |
response = requests.post(API_URL, headers=headers, json=payload)
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
return data["choices"][0]["text"].strip()
|
| 42 |
-
elif response.status_code == 429:
|
| 43 |
-
return "μμ² νλ μ΄κ³Ό: μ μ ν λ€μ μλν΄μ£ΌμΈμ."
|
| 44 |
-
else:
|
| 45 |
-
return f"μ€λ₯κ° λ°μνμ΅λλ€. μν μ½λ: {response.status_code}"
|
| 46 |
-
except Exception as e:
|
| 47 |
-
return f"μμ² μ²λ¦¬ μ€ μμΈκ° λ°μνμ΅λλ€: {str(e)}"
|
| 48 |
|
| 49 |
# Gradio μΈν°νμ΄μ€ ꡬμ±
|
| 50 |
with gr.Interface(fn=predict, inputs="text", outputs="text") as interface:
|
| 51 |
-
interface.launch()
|
|
|
|
| 17 |
sys.tracebacklimit = 0
|
| 18 |
|
| 19 |
def predict(inputs):
|
|
|
|
| 20 |
payload = {
|
| 21 |
+
"model": "gpt-3.5-turbo", # μμ λͺ¨λΈ, μ€μ μꡬ μ¬νμ λ§λ λͺ¨λΈ μλ³μλ‘ λ³κ²½ νμ
|
| 22 |
+
"messages": [{"role": "user", "content": inputs}], # chat/completions μλν¬μΈνΈμ λ§λ νμ΄λ‘λ ꡬ쑰
|
|
|
|
|
|
|
| 23 |
}
|
| 24 |
|
| 25 |
headers = {
|
|
|
|
| 27 |
"Authorization": f"Bearer {OPENAI_API_KEY}"
|
| 28 |
}
|
| 29 |
|
|
|
|
|
|
|
| 30 |
response = requests.post(API_URL, headers=headers, json=payload)
|
| 31 |
+
if response.status_code == 200:
|
| 32 |
+
data = response.json()
|
| 33 |
+
return data["choices"][0]["message"]["content"] # chat/completions μλ΅ κ΅¬μ‘°μ λ§μΆ° μ κ·Ό
|
| 34 |
+
else:
|
| 35 |
+
return f"μ€λ₯κ° λ°μνμ΅λλ€. μν μ½λ: {response.status_code}"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
# Gradio μΈν°νμ΄μ€ ꡬμ±
|
| 38 |
with gr.Interface(fn=predict, inputs="text", outputs="text") as interface:
|
| 39 |
+
interface.launch()
|