Update app.py

app.py

@@ -107,8 +107,8 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     skill_dict={}
     new_stat="*******************\n"
     for i,line in enumerate(lines):
-        if "Choices:" in line:
-            for z in range(1,10):
+        if ":  " in line:
+            for z in range(1,5):
                 try:
                     if f'{z}' in lines[i+z]:
                         print(lines[i+z].split(" ",1)[1])