Update app.py

app.py

@@ -72,9 +72,9 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     for response in stream:
         output += response.token.text
         if history:
-            yield "",[(prompt,output)]
+            yield "",[(prompt,output)],health
         else:
-            yield "",[(prompt,output)]
+            yield "",[(prompt,output)],health
 
     lines = output.strip().strip("\n").split("\n")
     #history=""