Update app.py
Browse files
app.py
CHANGED
@@ -1,40 +1,46 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
-
import
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
#
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# gradio 라이브러리를 gr 별칭으로 임포트합니다.
|
2 |
import gradio as gr
|
3 |
+
from gradio_client import Client
|
4 |
+
import httpx
|
5 |
+
|
6 |
+
# Client 객체 생성 시 timeout 매개변수 제거
|
7 |
+
client = Client("seawolf2357/sajuice")
|
8 |
+
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
# 타임아웃 값을 초 단위로 설정합니다. 예: 읽기 타임아웃을 30초로 설정
|
13 |
+
timeout_config = httpx.Timeout(5.0, read=30.0)
|
14 |
+
|
15 |
+
try:
|
16 |
+
# 수정된 타임아웃 설정을 적용하여 클라이언트를 생성합니다.
|
17 |
+
client = Client("arxivgpt/sajutest", timeout=timeout_config)
|
18 |
+
print("클라이언트가 성공적으로 생성되었습니다.")
|
19 |
+
except httpx.ReadTimeout:
|
20 |
+
print("요청 시간 초과: 서버가 지정된 시간 내에 응답하지 않았습니다.")
|
21 |
+
except Exception as e:
|
22 |
+
print(f"예상치 못한 오류가 발생했습니다: {e}")
|
23 |
+
|
24 |
+
|
25 |
+
def call_api(생년, 생월, 생일, 생시):
|
26 |
+
# API를 호출하고 응답값을 받습니다.
|
27 |
+
result = client.predict(
|
28 |
+
생년, # '생년(예: 1990)'에 대한 입력값
|
29 |
+
생월, # '생월(예: 01)'에 대한 입력값
|
30 |
+
생일, # '생일(예: 31)'에 대한 입력값
|
31 |
+
생시, # '생시(예시: 1030)'에 대한 입력값
|
32 |
+
api_name="/get_lunar_info"
|
33 |
+
)
|
34 |
+
return result
|
35 |
+
|
36 |
+
# Gradio 인터페이스를 정의합니다.
|
37 |
+
inputs = [
|
38 |
+
gr.Textbox(label="생년(예: 1990)"),
|
39 |
+
gr.Textbox(label="생월(예: 01)"),
|
40 |
+
gr.Textbox(label="생일(예: 31)"),
|
41 |
+
gr.Textbox(label="생시(예시: 1030)")
|
42 |
+
]
|
43 |
+
outputs = gr.Textbox(label="결과")
|
44 |
+
|
45 |
+
# 인터페이스를 생성하고 실행합니다.
|
46 |
+
gr.Interface(fn=call_api, inputs=inputs, outputs=outputs, title="Get Lunar Info").launch()
|