seawolf2357 commited on
Commit
3c19628
·
verified ·
1 Parent(s): 30072fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
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 응답에서 JSON을 파싱하여 결과를 추출합니다.
 
 
21
  try:
22
- generated_text = response.json().get('generated_text', '')
23
  except json.JSONDecodeError as e:
24
- print("Failed to decode JSON from response:", response.text)
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,