Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
import os
|
|
|
4 |
|
5 |
# 환경 변수에서 Hugging Face API 토큰을 가져옵니다.
|
6 |
ACCESS_TOKEN = os.getenv("HF_TOKEN")
|
@@ -13,18 +14,20 @@ def respond(message, max_tokens=512, temperature=0.7, top_p=0.95):
|
|
13 |
"temperature": temperature,
|
14 |
"top_p": top_p
|
15 |
}
|
16 |
-
|
17 |
-
# API 요청을
|
18 |
response = requests.post("http://hugpu.ai:7877/api/generate", json=data)
|
19 |
|
20 |
-
# API
|
|
|
|
|
21 |
try:
|
22 |
-
generated_text = response.json().get('generated_text', '')
|
23 |
except json.JSONDecodeError as e:
|
24 |
-
print("Failed to decode JSON from response:",
|
25 |
generated_text = "An error occurred while processing your request."
|
26 |
-
|
27 |
-
return generated_text
|
28 |
|
29 |
demo = gr.Interface(
|
30 |
fn=respond,
|
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
import os
|
4 |
+
import json # json 모듈 임포트 추가
|
5 |
|
6 |
# 환경 변수에서 Hugging Face API 토큰을 가져옵니다.
|
7 |
ACCESS_TOKEN = os.getenv("HF_TOKEN")
|
|
|
14 |
"temperature": temperature,
|
15 |
"top_p": top_p
|
16 |
}
|
17 |
+
|
18 |
+
# 외부 API로 요청을 보냄
|
19 |
response = requests.post("http://hugpu.ai:7877/api/generate", json=data)
|
20 |
|
21 |
+
# API 응답을 로그로 출력
|
22 |
+
print("API Response:", response.text)
|
23 |
+
|
24 |
try:
|
25 |
+
generated_text = response.json().get('generated_text', '') # API 응답에서 텍스트 추출
|
26 |
except json.JSONDecodeError as e:
|
27 |
+
print("Failed to decode JSON from response:", e)
|
28 |
generated_text = "An error occurred while processing your request."
|
29 |
+
|
30 |
+
return generated_text # 생성된 텍스트 반환
|
31 |
|
32 |
demo = gr.Interface(
|
33 |
fn=respond,
|