arxivgpt kim commited on
Commit
186f232
Β·
verified Β·
1 Parent(s): 7e30e77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -20
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-4-turbo-preview",
23
- "prompt": inputs,
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
- try:
38
- response = requests.post(API_URL, headers=headers, json=payload)
39
- if response.status_code == 200:
40
- data = response.json()
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()