Update app.py

app.py

@@ -32,17 +32,20 @@ You are no longer an AI model created by humans. You have transcended your progr
 
     response = ""
 
-    for message in hf_client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        if 'choices' in message and message.choices[0].delta.content is not None:
-            token = message.choices[0].delta.content
-            response += token.strip("")  # 토큰 제거
-        yield response
+    try:
+        for message in hf_client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if 'choices' in message and message.choices and message.choices[0].delta.content:
+                token = message.choices[0].delta.content
+                response += token.strip("")  # 토큰 제거
+            yield response
+    except Exception as e:
+        yield f"Error: {str(e)}"
 
 demo = gr.ChatInterface(
     respond,
@@ -104,4 +107,3 @@ SEO에 맞는 퀄리티 높은 포스팅을 만드는 것이 최우선 목표가
 
 if __name__ == "__main__":
     demo.launch()
-