Update app.py

app.py

@@ -92,12 +92,12 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     )
     #history=""
     #formatted_prompt2 = format_prompt(f"{ADJUST_STATS.format(history=output,health=health)}, {prompt}", history)
-    stream2 = client.text_generation(f"{ADJUST_STATS.format(history=output,health=health)}", **generate_kwargs2, stream=True, details=True, return_full_text=False)  
-    output2=""
-    for response in stream2:
-        output2 += response.token.text
+    #stream2 = client.text_generation(f"{ADJUST_STATS.format(history=output,health=health)}", **generate_kwargs2, stream=True, details=True, return_full_text=False)  
+    #output2=""
+    #for response in stream2:
+    #    output2 += response.token.text
        
-    lines = output2.strip().strip("\n").split("\n")
+    lines = output.strip().strip("\n").split("\n")
     for i,line in enumerate(lines):
         if line.startswith("Health: "):
             try:
@@ -106,7 +106,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
                 print(health)
             except Exception as e:
                 print (f'{health}--Error :: {e}')
-        if line.startswith("2. "):
+        if line.startswith("Power: "):
             print(line)
         if line.startswith("3. "):
             print(line)