ginipick commited on
Commit
dfe6166
·
verified ·
1 Parent(s): 71c328b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -525,7 +525,7 @@ def extract_image_prompt(blog_text: str, topic: str):
525
 
526
  try:
527
  response = client.chat.completions.create(
528
- model="gpt-3.5-turbo", # 일반적으로 사용 가능한 모델로 설정
529
  messages=[
530
  {"role": "system", "content": "Generate a single-line English image prompt from the following text. Return only the prompt text, nothing else."},
531
  {"role": "user", "content": f"Topic: {topic}\n\n---\n{blog_text}\n\n---"}
@@ -555,7 +555,7 @@ def ginigen_app():
555
 
556
  # Set default session state
557
  if "ai_model" not in st.session_state:
558
- st.session_state.ai_model = "gpt-3.5-turbo" # 기본 모델로 설정
559
  if "messages" not in st.session_state:
560
  st.session_state.messages = []
561
  if "auto_save" not in st.session_state:
@@ -576,7 +576,7 @@ def ginigen_app():
576
  sb.title("Blog Settings")
577
 
578
  # 모델 선택 추가
579
- available_models = ["gpt-3.5-turbo", "gpt-4o", "gpt-4-turbo"]
580
  sb.selectbox("AI Model", available_models, key="ai_model")
581
 
582
  sb.subheader("Blog Style Settings")
 
525
 
526
  try:
527
  response = client.chat.completions.create(
528
+ model="gpt-4.1-mini", # 일반적으로 사용 가능한 모델로 설정
529
  messages=[
530
  {"role": "system", "content": "Generate a single-line English image prompt from the following text. Return only the prompt text, nothing else."},
531
  {"role": "user", "content": f"Topic: {topic}\n\n---\n{blog_text}\n\n---"}
 
555
 
556
  # Set default session state
557
  if "ai_model" not in st.session_state:
558
+ st.session_state.ai_model = "gpt-4.1-mini" # 기본 모델로 설정
559
  if "messages" not in st.session_state:
560
  st.session_state.messages = []
561
  if "auto_save" not in st.session_state:
 
576
  sb.title("Blog Settings")
577
 
578
  # 모델 선택 추가
579
+ available_models = ["gpt-4.1-mini", "gpt-4o", "gpt-4-turbo"]
580
  sb.selectbox("AI Model", available_models, key="ai_model")
581
 
582
  sb.subheader("Blog Style Settings")