ginipick commited on
Commit
b2b3a97
·
verified ·
1 Parent(s): 11a6025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -37,11 +37,10 @@ from gradio_app.all_models import model_zoo
37
  # ===============================
38
  def text_to_image(height, width, steps, scales, prompt, seed):
39
  """
40
- 주어진 파라미터를 이용해 외부 API (http://211.233.58.201:7971/)의
41
- /process_and_save_image 엔드포인트를 호출하여 이미지를 생성한다.
42
  """
43
  from gradio_client import Client
44
- client = os.getenv("CLIENT_API")
45
  result = client.predict(
46
  height,
47
  width,
@@ -51,7 +50,6 @@ def text_to_image(height, width, steps, scales, prompt, seed):
51
  seed,
52
  api_name="/process_and_save_image"
53
  )
54
- # 결과가 dict이면 "url" 키의 값을, 아니라면 그대로 반환합니다.
55
  if isinstance(result, dict):
56
  return result.get("url", None)
57
  else:
@@ -59,13 +57,13 @@ def text_to_image(height, width, steps, scales, prompt, seed):
59
 
60
  def update_random_seed():
61
  """
62
- 외부 API의 /update_random_seed 엔드포인트를 호출하여
63
- 새로운 랜덤 시드 값을 가져온다.
64
  """
65
  from gradio_client import Client
66
- client = os.getenv("CLIENT_API")
67
  return client.predict(api_name="/update_random_seed")
68
 
 
69
  _TITLE = '''✨ 3D LLAMA Studio'''
70
  _DESCRIPTION = '''
71
  ### Welcome to 3D Llama Studio - Your Advanced 3D Generation Platform
 
37
  # ===============================
38
  def text_to_image(height, width, steps, scales, prompt, seed):
39
  """
40
+ 주어진 파라미터를 이용해 외부 API /process_and_save_image 엔드포인트를 호출하여 이미지를 생성한다.
 
41
  """
42
  from gradio_client import Client
43
+ client = Client(os.getenv("CLIENT_API")) # 기본값 설정
44
  result = client.predict(
45
  height,
46
  width,
 
50
  seed,
51
  api_name="/process_and_save_image"
52
  )
 
53
  if isinstance(result, dict):
54
  return result.get("url", None)
55
  else:
 
57
 
58
  def update_random_seed():
59
  """
60
+ 외부 API의 /update_random_seed 엔드포인트를 호출하여 새로운 랜덤 시드 값을 가져온다.
 
61
  """
62
  from gradio_client import Client
63
+ client = Client(os.getenv("CLIENT_API")) # 기본값 설정
64
  return client.predict(api_name="/update_random_seed")
65
 
66
+
67
  _TITLE = '''✨ 3D LLAMA Studio'''
68
  _DESCRIPTION = '''
69
  ### Welcome to 3D Llama Studio - Your Advanced 3D Generation Platform