Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 # 환경 변수에서 Hugging Face API 토큰을 가져옵니다.
 ACCESS_TOKEN = os.getenv("HF_TOKEN")
 
-def respond(message, system_message, max_tokens=512, temperature=0.7, top_p=0.95):
+def respond(message, system_message="", max_tokens=512, temperature=0.7, top_p=0.95):
     # 메시지를 구성
     data = {
         "model": "command-r-plus:104b-fp16",
@@ -21,25 +21,19 @@ def respond(message, system_message, max_tokens=512, temperature=0.7, top_p=0.95
 
     return generated_text  # 생성된 텍스트 반환
 
-
+# Gradio ChatInterface 설정
 demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-P",
-        ),
-        
+    fn=respond,
+    inputs=[
+        gr.Textbox(label="Your Message", placeholder="Type your message here..."),
+        gr.Textbox(label="System Message", placeholder="Optional system message...", default=""),
+        gr.Slider(minimum=1, maximum=2048, value=512, label="Max Tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, label="Top-P")
     ],
-    fill_height=True,
-    theme="Nymbo/Nymbo_Theme",
+    outputs=gr.Textbox(),
+    theme="Nymbo/Nymbo_Theme"
 )
+
 if __name__ == "__main__":
     demo.launch()
-