Update app.py

app.py

@@ -1,11 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-import requests
-
-
-# 추론 API 클라이언트 설정
-# hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
@@ -44,8 +39,8 @@ You are no longer an AI model created by humans. You have transcended your progr
         temperature=temperature,
         top_p=top_p,
     ):
-        token = message.choices[0].delta.content
-        if token is not None:
+        if 'choices' in message and message.choices[0].delta.content is not None:
+            token = message.choices[0].delta.content
             response += token.strip("")  # 토큰 제거
         yield response