Update app.py

app.py

@@ -63,6 +63,8 @@ def get_openai_generator(prompt: str):
         if "content" in event["choices"][0].delta:
             current_response = event["choices"][0].delta.content
             yield current_response
+        if "content" not in event["choices"][0].delta:
+            yield "[DONE]"
 
 @app.get("/")
 async def base_url():