Update app.py

app.py

@@ -344,6 +344,12 @@ def handsome_chat_completions():
                         ]
                     }
                     yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
+
+                    model_data = {
+                        "model_name": model_name
+                    }
+
+                    yield f"data: {json.dumps(model_data)}\n\n".encode("utf-8")
                     yield f"data: [DONE]\n\n".encode("utf-8")
                 return Response(
                     stream_with_context(generate_canned_stream()),
@@ -383,7 +389,13 @@ def handsome_chat_completions():
                             }
                         ]
                     }
+
+                    model_data = {
+                        "model_name": model_name
+                    }
+
                     yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
+                    yield f"data: {json.dumps(model_data)}\n\n".encode("utf-8")
                     yield f"data: [DONE]\n\n".encode("utf-8")
                 return Response(
                     stream_with_context(generate_canned_stream()),